MzeroMiko / VMamba

VMamba: Visual State Space Models,code is based on mamba
MIT License
1.82k stars 98 forks source link

I found that during debugging, I couldn't step into the `selective_scan_cuda_oflex` function, although it imports correctly and runs normally. Where can I find the specific details of the code for this function? #244

Open rkj12345 opened 1 week ago

MzeroMiko commented 1 week ago

This is implemented with cuda,which is compiled and then runned like a C programs, which is different from python. You can refer to the code on kernels/selective_scan for details.