FloMru

ImportError causal_conv1d_cuda.cpython-310-x86_64-linux-gnu.so undefined symbol

13

I am encountering a strange error, while using Mamba with torch 2.1.1+cu118 on Linux. In another environment with torch 1.13.1+cu116 the same code works fine. `ImportError: /home/user1/.conda/envs/2prod/lib/python3.10/site-packages/causal_conv1d_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda20CUDACachingAllocator12recordStreamERKNS_7DataPtrENS0_10CUDAStreamE`...

SequenceModel step function _setup_state missing?

1

I am building an encoder/decoder architecture where the encoder and decoder are S4 models. Since this is for a generative task I want to use the step function defined in...