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 10 months ago • 1 comments

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