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