Skip to content

Commit

Permalink
Update the NVTX bench helper for the new nsys utility (rapidsai#4826)
Browse files Browse the repository at this point in the history
Authors:
  - Victor Lafargue (https://github.com/viclafargue)

Approvers:
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: rapidsai#4826
  • Loading branch information
viclafargue authored Sep 1, 2022
1 parent 1cbcb63 commit cd33ec3
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/cuml/benchmark/nvtx_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

class Profiler:
def __init__(self, tmp_path='/tmp/nsys_report'):
self.qdrep_file = tmp_path + '/report.qdrep'
self.nsys_file = tmp_path + '/report.nsys-rep'
self.json_file = tmp_path + '/report.json'
self._execute(['rm', '-rf', tmp_path])
self._execute(['mkdir', '-p', tmp_path])
Expand All @@ -39,8 +39,8 @@ def _nsys_profile(self, command):
'profile',
'--trace=nvtx',
'--force-overwrite=true',
'--output={qdrep_file}'.format(
qdrep_file=self.qdrep_file)]
'--output={nsys_file}'.format(
nsys_file=self.nsys_file)]
profile_command.extend(command.split(' '))
self._execute(profile_command)

Expand All @@ -52,7 +52,7 @@ def _nsys_export2json(self):
'--force-overwrite=true',
'--output={json_file}'.format(
json_file=self.json_file),
self.qdrep_file]
self.nsys_file]
self._execute(export_command)

def _parse_json(self):
Expand Down

0 comments on commit cd33ec3

Please sign in to comment.