search

hustvl / Vim

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
Apache License 2.0
2.56k stars 160 forks source link

pip install -e mamba #20

Closed innocence0206 closed 4 months ago

innocence0206 commented 4 months ago

I got error: namespace "cub" has no member "WarpMask" when installing the mamba, but I have no idea how to solve it, could you give me some advice? Thanks

xuanhu001 commented 1 week ago

I met the same problem when I ran pip install -e mamba-1p1p1 in Vim. This is error log.

Vim/mamba-1p1p1/csrc/selective_scan/reverse_scan.cuh(120): error: namespace "cub" has no member "WarpMask"
              detected during:
                instantiation of "void BlockReverseScan<T, BLOCK_DIM_X, MEMOIZE>::ExclusiveReverseScan(T, T &, ScanOp, BlockPostfixCallbackOp &) [with T=float4, BLOCK_DIM_X=32, MEMOIZE=false, ScanOp=SSMScanOp<complex_t>, BlockPostfixCallbackOp=SSMScanPrefixCallbackOp<complex_t>]"

Could you share how to solve it? Thank you in advance for your time and assistance.