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 afeb356 commit 58283abCopy full SHA for 58283ab
tests/py/dynamo/automatic_plugin/test_flashinfer_rmsnorm.py
@@ -11,8 +11,10 @@
11
12
from ..conversion.harness import DispatchTestCase
13
14
-if importlib.util.find_spec("flashinfer"):
15
- import flashinfer
+# flashinfer has been impacted by torch upstream change: https://github.com/pytorch/pytorch/commit/660b0b8128181d11165176ea3f979fa899f24db1
+# 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
18
19
20
@torch.library.custom_op("flashinfer::rmsnorm", mutates_args=()) # type: ignore[misc]
0 commit comments