Kernels:
Trusted publisher
Add selective_scan_cuda binary for Torch 2.6, CUDA 12.4
Browse filesPre-compiled CUDA kernel for Mamba-SSM.
### Environment Specs:
- **PyTorch**: 2.6.0
- **CUDA**: 12.4
- **CXX11 ABI**: False
- **Architecture**: x86_64
- **Python**: 3.11
Works on Meluxina, the Luxembourguish supercomputer
build/torch26-cxx11abifalse-cu124-x86_64-linux/selective_scan_cuda.cpython-311-x86_64-linux-gnu.so
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31ecdf603f26f48fc811fb1e45ffa6250ab071a2d677565d1c454ab9ea600a5b
|
| 3 |
+
size 632325768
|