From c4cb4ca2e566906788aa103790172c0636dd830a Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Tue, 19 Nov 2024 16:54:52 -0500 Subject: [PATCH] Bugfix --- test/test_gpu/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_gpu/main.py b/test/test_gpu/main.py index a0012ec4..6e7a3626 100644 --- 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()