diff --git a/llvm/projects/gpu_profiler/offline_profiler.cpp b/llvm/projects/gpu_profiler/offline_profiler.cpp index 23543323df7f54c3c84bd53793876df43c7e9fbe..6b9f37ef62cc2c8600d11474100f27873bc36d7a 100644 --- a/llvm/projects/gpu_profiler/offline_profiler.cpp +++ b/llvm/projects/gpu_profiler/offline_profiler.cpp @@ -499,6 +499,7 @@ public: } void dumpTensorInfo(const char * const filename) const { + std::cout<<"dumping to"<<filename<<'\n'; const std::string header = "Op,Time (ms),Energy (mJ),GPU Energy (mJ),DDR Energy (mJ),Power (mW),GPU Power (mW),DDR Power (mW),Time std,Energy std,GPU Energy std,DDR Energy std,Power std,GPU Power std,DDR Power std\n"; std::ofstream ofs; ofs.open(filename); @@ -583,11 +584,11 @@ int main(int argc, char *argv[]) { i += 1; Profiler pp; - pp.profile(program, std::stoi(argv[i + 1])); - pp.dumpTensorInfo(argv[i + 2]); + pp.profile(program, std::stoi(argv[i])); + pp.dumpTensorInfo(argv[i + 1]); if (argc > NUM_ARGS) - pp.dumpPowerReadings(argv[i + 3]); + pp.dumpPowerReadings(argv[i + 2]); return 0; }