Skip to content

wangshanyw/PyTorch-CUDA-Operators-Implementation-and-Optimization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages