Added NVTX instrumentation