- PyTorch CUDA Extension Demo.
- Implements a sample Monte-Carlo integration routine of:
- Green's function for 2D Laplacian operator;
- Over a complex polygonal domain (without self intersection, with holes);
- With respect to a huge number of center points;
- Differentiable with respect to domain boundary.
- Fixes missing parts in libtorch's offical CMake config module for CUDA extensions
- in a brute-force manner.
axihixa / pte Goto Github PK
View Code? Open in Web Editor NEWPyTorch CUDA Extension Demo