From 2ffa6f4fd9cf8caa643adb5fb5fd57b7f7ba2dd1 Mon Sep 17 00:00:00 2001 From: Galaxy1458 <55453380+Galaxy1458@users.noreply.github.com> Date: Thu, 2 Jan 2025 10:42:50 +0800 Subject: [PATCH] fix bug (#384) --- src/flag_gems/runtime/backend/_nvidia/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/flag_gems/runtime/backend/_nvidia/__init__.py b/src/flag_gems/runtime/backend/_nvidia/__init__.py index b9c9788d..01f514f6 100644 --- 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" )