/pytorch_custom_cpp_kernel_example

An experiment of implementing a PyTorch custom kernel that dispatches operations to other computing nodes.

Primary LanguageC++

No issues in this repository yet.