Initial commit
Showing
- README.md 33 additions, 0 deletionsREADME.md
- SM-dispatch/SM-small-gemm.cu 178 additions, 0 deletionsSM-dispatch/SM-small-gemm.cu
- mlp_mps/cudastr_linear_ops.py 21 additions, 0 deletionsmlp_mps/cudastr_linear_ops.py
- mlp_mps/cudastr_linear_torchm.py 21 additions, 0 deletionsmlp_mps/cudastr_linear_torchm.py
- mlp_mps/custom_linear_gemm.cu 48 additions, 0 deletionsmlp_mps/custom_linear_gemm.cu
- mlp_mps/custom_linear_gemv.cu 39 additions, 0 deletionsmlp_mps/custom_linear_gemv.cu
- mlp_mps/custom_relu.cu 30 additions, 0 deletionsmlp_mps/custom_relu.cu
- mlp_mps/linear.py 48 additions, 0 deletionsmlp_mps/linear.py
- mlp_mps/linear_ops.py 47 additions, 0 deletionsmlp_mps/linear_ops.py
- mlp_mps/mlp_ops_batch.py 53 additions, 0 deletionsmlp_mps/mlp_ops_batch.py
- mlp_mps/mlp_ops_split.py 70 additions, 0 deletionsmlp_mps/mlp_ops_split.py
- mlp_mps/mlp_torch_batch.py 46 additions, 0 deletionsmlp_mps/mlp_torch_batch.py
- mlp_mps/mlp_torch_split.py 79 additions, 0 deletionsmlp_mps/mlp_torch_split.py
- mlp_mps/mlp_torch_split_pool.py 59 additions, 0 deletionsmlp_mps/mlp_torch_split_pool.py
- mlp_mps/ops.cpp 27 additions, 0 deletionsmlp_mps/ops.cpp
- mlp_mps/outputnewnew.csv 20 additions, 0 deletionsmlp_mps/outputnewnew.csv
- mlp_mps/run.sh 26 additions, 0 deletionsmlp_mps/run.sh
- mlp_mps/setup.py 20 additions, 0 deletionsmlp_mps/setup.py
- mlp_mps/testmps.py 31 additions, 0 deletionsmlp_mps/testmps.py
- mpstest/gemv.cu 24 additions, 0 deletionsmpstest/gemv.cu
Please register or sign in to comment