diff --git a/test/test_jit_fuser.py b/test/test_jit_fuser.py index a3f5ca00519f..d22c86c1ae38 100644 --- a/test/test_jit_fuser.py +++ b/test/test_jit_fuser.py @@ -31,7 +31,7 @@ def _test_fused_abs(self, device='cpu'): @torch.jit.script def func(x): - return x.abs() * 2 + return x.abs()*2 a = torch.randn(5, device=device) self.assertEqual(func(a), a.abs() * 2)