Skip to content

Commit 68c9365

Browse files
committed
remove timing util
1 parent 0282106 commit 68c9365

File tree

1 file changed

+0
-24
lines changed

1 file changed

+0
-24
lines changed

cuda_core/examples/jit_lto_fractal.py

Lines changed: 0 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -104,30 +104,6 @@ def link(self, user_code, target_type):
104104
return linked_code.get_kernel("main_workflow")
105105

106106
def run(self, kernel):
107-
from cuda.bindings import driver
108-
109-
e1 = e2 = None
110-
N = 10**4
111-
N_warmup = 10
112-
ave_time = 0.0
113-
for i in range(N + N_warmup):
114-
if e1 is None:
115-
e1 = self.stream.record(options={"enable_timing": True})
116-
else:
117-
self.stream.record(e1)
118-
launch(kernel, self.config, self.buffer.data.ptr)
119-
if e2 is None:
120-
e2 = self.stream.record(options={"enable_timing": True})
121-
else:
122-
self.stream.record(e2)
123-
e2.sync()
124-
if i < N_warmup:
125-
continue
126-
err, t = driver.cuEventElapsedTime(e1.handle, e2.handle)
127-
assert err == 0
128-
ave_time += t
129-
print("timing: ", ave_time / N)
130-
131107
launch(kernel, self.config, self.buffer.data.ptr)
132108
self.stream.sync()
133109

0 commit comments

Comments
 (0)