add lightllm ops

niushengxiao · niushengxiao · commit 815a698be46f · 2025-05-28T13:06:04.000+08:00
diff --git a/lightllm/utils/light_utils.py b/lightllm/utils/light_utils.py
@@ -5,7 +5,7 @@
     # TODO: lightllm_kernel release
     import lightllm_kernel
 
-    light_ops = lightllm_kernel
+    light_ops = getattr(lightllm_kernel, "ops", lightllm_kernel)
     HAS_LIGHTLLM_KERNEL = True
 except:
     light_ops = None
