File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
tests/py/dynamo/automatic_plugin Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 11
11
12
12
from ..conversion .harness import DispatchTestCase
13
13
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
18
18
19
19
20
20
@torch .library .custom_op ("flashinfer::rmsnorm" , mutates_args = ()) # type: ignore[misc]
You can’t perform that action at this time.
0 commit comments