NX-AI / xlstm

Official repository of the xLSTM.
GNU Affero General Public License v3.0
918 stars 66 forks source link

RuntimeError: Error building extension 'slstm_HS128BS8NH4NS4DBfDRbDWbDGbDSbDAfNG4SA1GRCV0GRC0d0FCV0FC0d0' #19

Closed qiwang067 closed 2 weeks ago

qiwang067 commented 3 weeks ago
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/block.py", line 32, in __init__
    super().__init__(
  File "/home/ubuntu/xlstm/xlstm/blocks/xlstm_block.py", line 62, in __init__
    self.xlstm = sLSTMLayer(config=self.config.slstm)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/layer.py", line 78, in __init__
    self.slstm_cell = sLSTMCell(self.config)
                      ^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/cell.py", line 780, in __new__
    return sLSTMCell_cuda(config, skip_backend_init=skip_backend_init)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/cell.py", line 690, in __init__
    self.func = sLSTMCellFuncGenerator(self.training, config)
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/cell.py", line 536, in sLSTMCellFuncGenerator
    slstm_cuda = sLSTMCellCUDA.instance(config=config)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/cell.py", line 515, in instance
    cls.mod[repr(config)] = load(
                            ^^^^^
  File "/home/ubuntu/xlstm/xlstm/blocks/slstm/src/cuda_init.py", line 84, in load
    mod = _load(name + suffix, sources, **myargs)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ubuntu/anaconda3/envs/xlstm/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1306, in load
    return _jit_compile(
           ^^^^^^^^^^^^^
  File "/home/ubuntu/anaconda3/envs/xlstm/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1710, in _jit_compile
    _write_ninja_file_and_build_library(
  File "/home/ubuntu/anaconda3/envs/xlstm/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1823, in _write_ninja_file_and_build_library
    _run_ninja_build(
  File "/home/ubuntu/anaconda3/envs/xlstm/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 2112, in _run_ninja_build
    raise RuntimeError(message) from e
RuntimeError: Error building extension 'slstm_HS128BS8NH4NS4DBfDRbDWbDGbDSbDAfNG4SA1GRCV0GRC0d0FCV0FC0d0'
liuzeyu6 commented 3 weeks ago

I've also run into the same problem.

louyongjun commented 3 weeks ago

conda install cccl

RayShark commented 3 weeks ago

same problem, any solution now?

qiwang067 commented 2 weeks ago

conda install cccl

I tried this solution, and it worked.

Marco-Nguyen commented 2 weeks ago

conda install cccl

@louyongjun What are your installation steps? I tried your solution but it doesn't work