Skip to content

Commit

Permalink
simplified and formatted with black
Browse files Browse the repository at this point in the history
  • Loading branch information
benvanwerkhoven committed Apr 22, 2024
1 parent bd0a40c commit dfd3da9
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions kernel_tuner/observers/register.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,14 @@
from kernel_tuner.observers.observer import BenchmarkObserver


class RegisterObserver(BenchmarkObserver):
"""Observer for counting the number of registers."""

def __init__(self) -> None:
super().__init__()

def get_results(self):
try:
registers_per_thread = self.dev.num_regs
except AttributeError:
raise NotImplementedError(f"Backend '{type(self.dev).__name__}' does not support count of registers per thread")
return {
"num_regs": registers_per_thread
}
raise NotImplementedError(
f"Backend '{type(self.dev).__name__}' does not support count of registers per thread"
)
return {"num_regs": registers_per_thread}

0 comments on commit dfd3da9

Please sign in to comment.