mamba icon indicating copy to clipboard operation
mamba copied to clipboard

Published 20 hours ago • verified publisher icon state-spaces

revise the selective_scan_cuda

Open Sunzechen1219 opened this issue 1 year ago • 1 comments
trafficstars

Hello！I want to change this function, but it looks like it's encapsulated by cuda_c, how do I change it?

Sunzechen1219 avatar Apr 23 '24 02:04 Sunzechen1219

You can change the CUDA code in csrc

tridao avatar Apr 23 '24 04:04 tridao

You can change the CUDA code in csrc

Could you specify which script in scrc needs to be modified, for example, if I want to revise selective_scan_cuda.fwd, which script and which part should I revise?

Thanks for replying.

EddieEduardo avatar Oct 24 '24 03:10 EddieEduardo