Skip to content

Commit

Permalink
Rename to prologue_start and prologue_end
Browse files Browse the repository at this point in the history
  • Loading branch information
csbnw committed Apr 24, 2024
1 parent 397306f commit 8061239
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
4 changes: 2 additions & 2 deletions kernel_tuner/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,10 +349,10 @@ def benchmark_default(self, func, gpu_args, threads, grid, result):
]

for obs in self.prologue_observers:
obs.before_start()
obs.prologue_start()
self.dev.run_kernel(func, gpu_args, threads, grid)
self.dev.synchronize()
obs.after_finish()
obs.prologue_finish()

self.dev.synchronize()
for _ in range(self.iterations):
Expand Down
4 changes: 2 additions & 2 deletions kernel_tuner/observers/ncu.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,10 +31,10 @@ def __init__(self, metrics=None, device=0):
self.device = device
self.results = dict()

def before_start(self):
def prologue_start(self):
nvmetrics.measureMetricsStart(self.metrics, self.device)

def after_finish(self):
def prologue_finish(self):
self.results = nvmetrics.measureMetricsStop()

def get_results(self):
Expand Down
8 changes: 4 additions & 4 deletions kernel_tuner/observers/observer.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,11 +61,11 @@ class PrologueObserver(BenchmarkObserver):
"""Observer that measures something in a seperate kernel invocation prior to the normal benchmark."""

@abstractmethod
def before_start(self):
"""before start is called before the kernel starts"""
def prologue_start(self):
"""prologue start is called before the kernel starts"""
pass

@abstractmethod
def after_finish(self):
"""after finish is called after the kernel has finished execution"""
def prologue_finish(self):
"""prologue finish is called after the kernel has finished execution"""
pass

0 comments on commit 8061239

Please sign in to comment.