Bugfix

pytorch-labs · Nov 20, 2024 · c4cb4ca · c4cb4ca
1 parent 5bfe61c
commit c4cb4ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_gpu/main.py b/test/test_gpu/main.py
@@ -88,9 +88,9 @@ def _run_operator_in_task(op: str, args: List[str]):
     task.make_operator_instance(args=args)
     task.run()
     task.check_output()
-    task.del_op_instance()
     # Test backward (if applicable)
     if task.get_attribute("has_bwd"):
+        task.del_op_instance()
         args.extend(["--bwd"])
         task.make_operator_instance(args=args)
         task.run()