fix bug (FlagOpen#384)

tianxiao-baai · Jan 2, 2025 · 2ffa6f4 · 2ffa6f4
1 parent e02af1f
commit 2ffa6f4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/flag_gems/runtime/backend/_nvidia/__init__.py b/src/flag_gems/runtime/backend/_nvidia/__init__.py
@@ -3,6 +3,8 @@
 from .heuristics_config_utils import HEURISTICS_CONFIGS
 
 global specific_ops, unused_ops
+specific_ops = None
+unused_ops = None
 vendor_info = VendorInfoBase(
     vendor_name="nvidia", device_name="cuda", device_query_cmd="nvidia-smi"
 )