I am trying to write a extension for pytorch. I find some repos to study. Now I have some problems:
- How about the relation between
module
andfunction
? I find some examples callfunction
inmodule
,why not only usemodule
to implement? - I find some cuda implement use
kernle
, and others not. When should I write the cuda kernel files ?