https://github.com/xgqdut2016/hpc2torch
https://github.com/xgqdut2016/hpc2torch
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/xgqdut2016/hpc2torch
- Owner: xgqdut2016
- Created: 2024-07-25T07:04:17.000Z (9 months ago)
- Default Branch: main
- Last Pushed: 2024-11-06T05:45:52.000Z (5 months ago)
- Last Synced: 2024-11-06T06:29:45.812Z (5 months ago)
- Language: Cuda
- Size: 31.3 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
- awesome-cuda-and-hpc - xgqdut2016/hpc2torch
README
# hpc2Torch
这个仓库打算搭建一个高性能底层库的测试框架，将会针对onnx的算子编写相关的高性能kernel，作为pytorch的补充，从python端对比手写kernel和pytorch库函数的性能以及精度对比。
## src
这个文件夹下面存放的是不同算子的kernel
## test
这个文件夹存放的是不同算子的python测试脚本，其中performance.py是功能文件，用于对比性能
## run.sh
默认编译CPU端代码，运行仓库命令是：
bash run.sh
编译结束以后，可以直接做python端测试，测试softmax算子的CPU端代码命令为：
python test/test_softmax.py --device cpu
如果需要编译测试其他平台代码，比如说GPU端测试，那么修改run.sh里面的cmake ../ -DUSE_CPU=ON为 cmake ../ -DUSE_CUDA=ON，对应的测试python脚本--device cpu也修改为--device cuda