Write my own extension for pytorch

If I want to implement some functions running on GPU , which , however, cannot completely use the official function and tensor to express, Should I have to write the cuda file by myself? @smth

(you dont need to tag me in your question, dont do that).

Yes write your own files. See https://github.com/pytorch/extension-ffi

1 Like