Skip to content

Commit 3602286

Browse files
committed
Raise exception when using NCUObserver without nvmetrics
1 parent 4cc68c5 commit 3602286

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

kernel_tuner/observers/ncu.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,17 @@ class NCUObserver(PrologueObserver):
2525

2626
def __init__(self, metrics=None, device=0):
2727
if not nvmetrics:
28-
print("NCUObserver is not available.")
28+
raise Exception("NCUObserver is not available.")
2929

3030
self.metrics = metrics
3131
self.device = device
3232
self.results = dict()
3333

3434
def before_start(self):
35-
if nvmetrics:
36-
nvmetrics.measureMetricsStart(self.metrics, self.device)
35+
nvmetrics.measureMetricsStart(self.metrics, self.device)
3736

3837
def after_finish(self):
39-
if nvmetrics:
40-
self.results = nvmetrics.measureMetricsStop()
38+
self.results = nvmetrics.measureMetricsStop()
4139

4240
def get_results(self):
4341
return dict(zip(self.metrics, self.results))

0 commit comments

Comments
 (0)