Add main

xuzhao9 · xuzhao9 · commit 59bb8fecbf9c · 2024-11-18T18:47:06.000-05:00
diff --git a/test/test_gpu/main.py b/test/test_gpu/main.py
@@ -102,9 +102,8 @@ def _run_operator_in_task(op: str, args: List[str]):
         task.run()
         task.check_output()
     except NotImplementedError:
-        logger.info(
-            f"Operator {op} does not support backward, skipping backward test."
-        )
+        # Operator does not support backward, skip the test
+        pass
 
 
 def make_test(operator):
