Hi @glaringlee!
Thanks for taking the time to answer my question!
I just want to check if this is valid:

Time the original function.
Make modifications to code.
Rebuild (python setup.py develop --cmake  along with some flags like BUILD_CAFFE2_OPS=0).
Test (python run_test.py for the specific module) to ensure correctness.
Time the function again (now that it’s been changed).