frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix lint in test_tensorexpr_pybind (#57869)

Browse Source 
Summary:
Fixes #{issue number}

Pull Request resolved: https://github.com/pytorch/pytorch/pull/57869

Pulled By: driazati

Reviewed By: gmagogsfm

Differential Revision: D28303542

fbshipit-source-id: a4c77c1e7ee92c39fd1c7d88422728e2f1c31680
This commit is contained in:
davidriazati@fb.com
2021-05-07 15:56:57 -07:00
committed by Facebook GitHub Bot
parent 4fad8d1a2c
commit ee4be5322b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_tensorexpr_pybind.py
View File

@ -255,7 +255,7 @@ graph(%a.1 : Float(3, 4, strides=[4, 1], requires_grad=0, device=cpu)):
@unittest.skipIf(not LLVM_ENABLED, "LLVM backend not enabled")
def test_kernel_with_permute(self):
def f(a):
return a.permute([2,1,0])
return a.permute([2, 1, 0])
device, size = 'cpu', (3, 4, 5)
x = torch.rand(size, device=device)
@ -281,10 +281,10 @@ graph(%a.1 : Float(3, 4, 5, strides=[20, 5, 1], requires_grad=0, device=cpu)):
@unittest.skipIf(not LLVM_ENABLED, "LLVM backend not enabled")
def test_kernel_with_expand(self):
def f(a):
return a.expand((2,3,4))
return a.expand((2, 3, 4))
device = 'cpu'
x = torch.rand((1,3,1), device=device)
x = torch.rand((1, 3, 1), device=device)
graph_str = """
graph(%a : Float(1, 3, 1, strides=[3, 1, 1], requires_grad=0, device=cpu)):
%1 : int = prim::Constant[value=2]()