Add main

pytorch-labs · Nov 18, 2024 · 59bb8fe · 59bb8fe
1 parent 6c8c122
commit 59bb8fe
Showing 1 changed file with 2 additions and 3 deletions.
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):