NameError: name 'selective_scan_cuda' is not defined

[YangJae96]

Hi.
I followed the instruction and cd kernels/selective_scan && pip install .

but when I run the code, it gets following error.

File "/root/workplace/deeplearning/VMamba/classification/models/csms6s.py", line 89, in forward
out, x, *rest = selective_scan_cuda.fwd(u, delta, A, B, C, D, None, delta_bias, delta_softplus)
NameError: name 'selective_scan_cuda' is not defined

It seems i cannot import selective_scan_cuda

import selective_scan_cuda
-> ModuleNotFoundError: No module named 'selective_scan_cuda'

How should I solve the problem?

I looked others question but cant figure it out.

Thanks in advance.