From bb79fae23ce1b42aeabc018a0edca00bcb9003d6 Mon Sep 17 00:00:00 2001 From: Michael Suo Date: Fri, 6 Sep 2019 16:17:41 -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..33357962e3d6 100644 --- a/test/test_jit_fuser.py +++ b/test/test_jit_fuser.py @@ -124,7 +124,7 @@ def test_checks_cat_inputs(self): # run the kernel. @torch.jit.script def f(x, y): - return torch.cat([x + 2 * x + x ** 2, y + 4 * y + y ** 3], dim=0) + return torch.cat([x + 2*x+x ** 2, y + 4 * y + y ** 3], dim=0) # NOTE: y is broadcastable to x, but output of f(x, y) should have # shape 3x4, and not 4x4.