[TEST] Temporary disable fp16 type_as test for PyTorch Frontend (apac…

…he#5799)
neo-ai · Jun 18, 2020 · 8eb41c1 · 8eb41c1
1 parent 7479041
commit 8eb41c1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py
@@ -842,13 +842,13 @@ def forward(self, *args):
 def test_type_as():
     torch.set_grad_enabled(False)
     input_shape = [1, 3]
-    
+
     def _create_module(dtype):
         class TypeAs(Module):
             def forward(self, *args):
                 expected_type_tensor = torch.zeros(1, 3, dtype=dtype)
                 return args[0].type_as(expected_type_tensor)
-        
+
         return TypeAs()
 
     input_data = torch.randn(input_shape).float()
@@ -868,7 +868,10 @@ def forward(self, *args):
         except Exception as e:
             # If GPU is not enabled in TVM, skip the fp16 test.
             pass
-
+
+        # Temporary disable fp16 test
+        check_fp16 = False
+
         if check_fp16:
             verify_model(_create_module(torch.float16), input_data=input_data)