From 75687d10f9fd690f4ed4b33361567dc9eb9aa501 Mon Sep 17 00:00:00 2001 From: Michael Suo Date: Fri, 6 Sep 2019 16:28:08 -0700 Subject: [PATCH] Update test_jit_fuser.py --- test/test_jit_fuser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_jit_fuser.py b/test/test_jit_fuser.py index d99dea7b3401..a3f5ca00519f 100644 --- a/test/test_jit_fuser.py +++ b/test/test_jit_fuser.py @@ -195,7 +195,7 @@ def test_chunk_correctness_cuda(self): def test_chunk_distributes_cuda(self): def f(x, y): z1, z2 = (x + y).chunk(2, dim=1) - return z1 * z2 + return z1*z2 x = torch.randn(4, 4, dtype=torch.float, device='cuda') y = torch.randn(4, 4, dtype=torch.float, device='cuda')