How to expose the "register_pre_hook()" interface?

[letian-zhang]

Hi, I stack in the first step. How to profile the runtime of forward and backward? I've never tried to modified the pytorch source code before, and the 'python_cpp_function.h' or 'python_cpp_function.cpp' isn't clear for me to write the pre_hook interface. Could you provide the change you made to pytorch source code? Thx a lot.

[deepakn94]

This is a good starting point: https://github.com/msr-fiddle/pipedream/blob/master/pre_hook.patch.

However, you shouldn't need to modify PyTorch source code if all you want to do is profile some models. Look at examples in profiler to see the application changes needed.