Writing custom cuda kernels for pytorch

1

I am trying to write custom cuda kernel for pytorch for a specific computation. Is there any available documentation for writing custom cuda kernels for pytorch?

2

There’s a gist that demonstrates how could you use PyCuda for that.

1 Like