Skip to content

Commit dfd3da9

Browse files
simplified and formatted with black
1 parent bd0a40c commit dfd3da9

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

kernel_tuner/observers/register.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,14 @@
11
from kernel_tuner.observers.observer import BenchmarkObserver
22

3+
34
class RegisterObserver(BenchmarkObserver):
45
"""Observer for counting the number of registers."""
56

6-
def __init__(self) -> None:
7-
super().__init__()
8-
97
def get_results(self):
108
try:
119
registers_per_thread = self.dev.num_regs
1210
except AttributeError:
13-
raise NotImplementedError(f"Backend '{type(self.dev).__name__}' does not support count of registers per thread")
14-
return {
15-
"num_regs": registers_per_thread
16-
}
11+
raise NotImplementedError(
12+
f"Backend '{type(self.dev).__name__}' does not support count of registers per thread"
13+
)
14+
return {"num_regs": registers_per_thread}

0 commit comments

Comments
 (0)