Skip to content

Commit 58283ab

Browse files
skip flashinfer test due to torch upstream change (#3794)
1 parent afeb356 commit 58283ab

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/py/dynamo/automatic_plugin/test_flashinfer_rmsnorm.py

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

1212
from ..conversion.harness import DispatchTestCase
1313

14-
if importlib.util.find_spec("flashinfer"):
15-
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
1618

1719

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

0 commit comments

Comments
 (0)