1. modified jit default args parser; 2. added test for flip() on empty tensor

weiyangfb · weiyangfb · commit fd1eff5bf5bd · 2018-07-23T19:52:07.000-07:00
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -6722,6 +6722,10 @@ def _test_flip(self, use_cuda=False):
         self.assertEqual(flip0_result, data.flip(0))
         self.assertEqual(flip1_result, data.flip(1))
 
+        # test empty tensor, should just return an empty tensor of the same shape
+        data = torch.tensor([])
+        self.assertEqual(data, data.flip(0))
+
     def test_flip(self):
         self._test_flip(self, use_cuda=False)
 
diff --git a/tools/jit/gen_jit_dispatch.py b/tools/jit/gen_jit_dispatch.py
@@ -419,7 +419,9 @@ def format_arg(arg):
                 .replace('false', 'False') \
                 .replace('nullptr', 'None') \
                 .replace('Reduction::ElementwiseMean', 'ElementwiseMean') \
-                .replace('{}', 'None' if is_tensor_arg(arg) else '[]')
+                .replace('{}', 'None' if is_tensor_arg(arg) else '[]') \
+                .replace('{', '[') \
+                .replace('}', ']')
 
             default = default_map.get(default, default)
             decl = '{}={}'.format(decl, default)
