Undefined symbol during import

[roastduck]

When import, the following error occurs.

Traceback (most recent call last):
  File "tmp.py", line 2, in <module>
    import torch_blocksparse
  File "/home/rd/.local/lib/python3.7/site-packages/torch_blocksparse/__init__.py", line 2, in <module>
    from .matmul import MatMul, Linear
  File "/home/rd/.local/lib/python3.7/site-packages/torch_blocksparse/matmul.py", line 4, in <module>
    import torch_blocksparse_cpp_utils
ImportError: /home/rd/.local/lib/python3.7/site-packages/torch_blocksparse_cpp_utils.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN3c104impl23ExcludeDispatchKeyGuardC1ENS_11DispatchKeyE

I am using PyTorch 1.7.0, and it seems this symbol is related to upgrading to PyTorch 1.7.0. However, PyTorch 1.6.0 doesn't work with the latest triton.