Skip to content

Commit 3402808

Browse files
committed
Rename to prologue_start and prologue_finish
1 parent 397306f commit 3402808

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

kernel_tuner/core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -349,10 +349,10 @@ def benchmark_default(self, func, gpu_args, threads, grid, result):
349349
]
350350

351351
for obs in self.prologue_observers:
352-
obs.before_start()
352+
obs.prologue_start()
353353
self.dev.run_kernel(func, gpu_args, threads, grid)
354354
self.dev.synchronize()
355-
obs.after_finish()
355+
obs.prologue_finish()
356356

357357
self.dev.synchronize()
358358
for _ in range(self.iterations):

kernel_tuner/observers/ncu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@ def __init__(self, metrics=None, device=0):
3131
self.device = device
3232
self.results = dict()
3333

34-
def before_start(self):
34+
def prologue_start(self):
3535
nvmetrics.measureMetricsStart(self.metrics, self.device)
3636

37-
def after_finish(self):
37+
def prologue_finish(self):
3838
self.results = nvmetrics.measureMetricsStop()
3939

4040
def get_results(self):

kernel_tuner/observers/observer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,11 +61,11 @@ class PrologueObserver(BenchmarkObserver):
6161
"""Observer that measures something in a seperate kernel invocation prior to the normal benchmark."""
6262

6363
@abstractmethod
64-
def before_start(self):
65-
"""before start is called before the kernel starts"""
64+
def prologue_start(self):
65+
"""prologue start is called before the kernel starts"""
6666
pass
6767

6868
@abstractmethod
69-
def after_finish(self):
70-
"""after finish is called after the kernel has finished execution"""
69+
def prologue_finish(self):
70+
"""prologue finish is called after the kernel has finished execution"""
7171
pass

0 commit comments

Comments
 (0)