PyTorch CUDA operators modifications
- Matrix multiplication and Softmax
- The cuda_operators/Aten folder is the modified code section, which corresponds to the pytorch/aten/src/Aten folder in the code framework.
Test files
- test_softmax.py
- test.py
PyTorch算子修改内容
- 矩阵乘法和Softmax
- 其中cuda_operators/Aten文件夹为修改代码部分,对应代码框架中的pytorch/aten/src/Aten文件夹
测试文件
- test_softmax.py
- test.py