I am trying to profile all the addition or multiplication computations completed at the very bottom operation level (two input operands and the result, for example, 2 + 3 = 5) during the inference of a model.
Is it possible to achieve this in python (such as the pytorch profiler) or I need to look into the C++ back end? (Will also need some hints if need to go deep into the C++)
Thank you!