NameError: name ‘selective_scan_cuda‘ is not defined

florent1no

已于 2025-01-07 15:59:41 修改

阅读量1.7k

点赞数 5

分类专栏： VMamba 文章标签： python 深度学习计算机视觉

于 2025-01-07 15:59:02 首次发布

本文链接：https://blog.youkuaiyun.com/m0_47694179/article/details/144987297

版权

VMamba 专栏收录该内容

3 篇文章

订阅专栏

VMamba代码中，在成功执行完依赖安装命令后，

cd kernels/selective_scan && pip install

执行代码报错：

File "xxxx/VMamba/classification/models/csms6s.py", line 85, 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

selective_scan相关依赖已经安装成功，检查依赖安装命令执行后生成的文件夹，其中不存在selective_scan_cuda，所以在csms6s.py文件中注释掉“selective_scan_cuda”相关代码，就可成功运行。

# if backend == "oflex":
#     out, x, *rest = selective_scan_cuda_oflex.fwd(u, delta, A, B, C, D, delta_bias, delta_softplus, 1, oflex)
# elif backend == "core":
#     out, x, *rest = selective_scan_cuda_core.fwd(u, delta, A, B, C, D, delta_bias, delta_softplus, 1)
# elif backend == "mamba":
# out, x, *rest = selective_scan_cuda.fwd(u, delta, A, B, C, D, None, delta_bias, delta_softplus)
out, x, *rest = selective_scan_cuda_oflex.fwd(u, delta, A, B, C, D, delta_bias, delta_softplus, 1, oflex)

后向传播部分做相应改动即可。