We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4cc68c5 commit 3602286Copy full SHA for 3602286
kernel_tuner/observers/ncu.py
@@ -25,19 +25,17 @@ class NCUObserver(PrologueObserver):
25
26
def __init__(self, metrics=None, device=0):
27
if not nvmetrics:
28
- print("NCUObserver is not available.")
+ raise Exception("NCUObserver is not available.")
29
30
self.metrics = metrics
31
self.device = device
32
self.results = dict()
33
34
def before_start(self):
35
- if nvmetrics:
36
- nvmetrics.measureMetricsStart(self.metrics, self.device)
+ nvmetrics.measureMetricsStart(self.metrics, self.device)
37
38
def after_finish(self):
39
40
- self.results = nvmetrics.measureMetricsStop()
+ self.results = nvmetrics.measureMetricsStop()
41
42
def get_results(self):
43
return dict(zip(self.metrics, self.results))
0 commit comments