We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5e14cb1 commit e44c707Copy full SHA for e44c707
vllm_ascend/utils.py
@@ -214,10 +214,12 @@ def enable_custom_op():
214
if _CUSTOM_OP_ENABLED is not None:
215
return _CUSTOM_OP_ENABLED
216
try:
217
+ # isort: off
218
# register custom ops into torch_library here
219
+ import vllm_ascend.vllm_ascend_C # type: ignore # noqa: F401
220
# register the meta implementation for custom kernel if necessary
221
import vllm_ascend.meta_registration # type: ignore # noqa: F401
- import vllm_ascend.vllm_ascend_C # type: ignore # noqa: F401
222
+ # isort: on
223
_CUSTOM_OP_ENABLED = True
224
except ImportError:
225
_CUSTOM_OP_ENABLED = False
0 commit comments