Skip to content

Commit 1681730

Browse files
committed
Added useful error message in case Register Observer is not supported
1 parent 943b3c4 commit 1681730

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

kernel_tuner/observers/register.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,10 @@ def __init__(self) -> None:
77
super().__init__()
88

99
def get_results(self):
10+
try:
11+
registers_per_thread = self.dev.num_regs
12+
except AttributeError:
13+
raise NotImplementedError(f"Backend '{type(self.dev).__name__}' does not support count of registers per thread")
1014
return {
11-
"num_regs": self.dev.num_regs
15+
"num_regs": registers_per_thread
1216
}

0 commit comments

Comments
 (0)