├── .gitignore ├── LICENSE ├── README-CN.md ├── README.md ├── csrc ├── add.cu ├── ops.h └── pybind.cpp ├── examples └── example_add.py ├── image └── Logo.png ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/LICENSE -------------------------------------------------------------------------------- /README-CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/README-CN.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/README.md -------------------------------------------------------------------------------- /csrc/add.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/csrc/add.cu -------------------------------------------------------------------------------- /csrc/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/csrc/ops.h -------------------------------------------------------------------------------- /csrc/pybind.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/csrc/pybind.cpp -------------------------------------------------------------------------------- /examples/example_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/examples/example_add.py -------------------------------------------------------------------------------- /image/Logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/image/Logo.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch == 2.1.2 2 | pybind11 3 | setuptools -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunkx109/My-Torch-Extension/HEAD/setup.py --------------------------------------------------------------------------------