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 943b3c4 commit 1681730Copy full SHA for 1681730
kernel_tuner/observers/register.py
@@ -7,6 +7,10 @@ def __init__(self) -> None:
7
super().__init__()
8
9
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")
14
return {
- "num_regs": self.dev.num_regs
15
+ "num_regs": registers_per_thread
16
}
0 commit comments