#C++扩展

extension-cpp项目演示了为PyTorch开发C++/CUDA扩展的方法，包括实现自定义CPU和CUDA内核的mymuladd运算符。项目提供构建和测试指南，以及Python、C++和CUDA版本的性能对比。适用于PyTorch 2.4+，是学习PyTorch扩展开发的实践示例。