Skip to content

Commit 0eacd3e

Browse files
committed
skip import flashinfer
1 parent 0e74828 commit 0eacd3e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/py/dynamo/automatic_plugin/test_flashinfer_rmsnorm.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@
1111

1212
from ..conversion.harness import DispatchTestCase
1313

14-
if importlib.util.find_spec("flashinfer"):
15-
# flashinfer has been impacted by torch upstream change: https://github.com/pytorch/pytorch/commit/660b0b8128181d11165176ea3f979fa899f24db1
16-
# got ImportError: cannot import name '_get_pybind11_abi_build_flags' from 'torch.utils.cpp_extension'
17-
# import flashinfer
14+
# flashinfer has been impacted by torch upstream change: https://github.com/pytorch/pytorch/commit/660b0b8128181d11165176ea3f979fa899f24db1
15+
# got ImportError: cannot import name '_get_pybind11_abi_build_flags' from 'torch.utils.cpp_extension'
16+
# if importlib.util.find_spec("flashinfer"):
17+
# import flashinfer
1818

1919

2020
@torch.library.custom_op("flashinfer::rmsnorm", mutates_args=()) # type: ignore[misc]

0 commit comments

Comments
 (0)