kwea123/pytorch_cppcuda_practice

This is stars and forks stats for /kwea123/pytorch_cppcuda_practice repository. As of 05 May, 2024 this repository has 18 stars and 0 forks.

pytorch_cppcuda_practice Practice to write cpp/cuda extension for pytorch This simple example writes a custom matrix addition in cuda, and provides python binding with forward/backward operations. Sources: 1 2 Installation Install pytorch. From this repo, run pip install . --use-feature=in-tree-build Test Run python test.py. Misc For vscode to resolve <torch/extension.h>, add the following lines to the include path (change python path and version): "<path to your conda environment>/include/python3.8", "<path...