GitHub - affjljoo3581/Experiments-PyTorch-CUDA-Extension-for-Sparse-MatMul: An example repository of PyTorch BlockSparse implementation with custom CUDA extension

affjljoo3581 / Experiments-PyTorch-CUDA-Extension-for-Sparse-MatMul Public

An example repository of PyTorch BlockSparse implementation with custom CUDA extension

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 258 Commits
native		native
sparse		sparse
.gitignore		.gitignore
bank_conflict_simulator.py		bank_conflict_simulator.py
benchmark.png		benchmark.png
sparse_matmul_gpu.cu		sparse_matmul_gpu.cu
sparse_softmax.cu		sparse_softmax.cu