mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

add option to only benchmark schedule [run_process_replay] (#6204)

Browse source
This commit is contained in:
qazal 2024-08-20 21:51:27 +08:00 • committed by GitHub
commit 074cf780dd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
test/external/external_benchmark_schedule.py vendored
View file

@ -10,6 +10,7 @@ if __name__ == "__main__":
PROFILE = getenv("PROFILE", 1)
FORWARD_ONLY = getenv("FORWARD_ONLY", 0)
SCHEDULE_ONLY = getenv("SCHEDULE_ONLY", 0)
with Profiling(PROFILE):
with Timing("***** model forward in "):
@ -20,18 +21,19 @@ if __name__ == "__main__":
with Timing("***** model schedule in "):
sched = out.schedule()
asts = {x.ast.key:x.ast for x in sched if x.ast.op is UOps.SINK}.values()
kernels = []
with Profiling(PROFILE):
with Timing("***** model uops in "):
for ast in asts:
k = Kernel(ast)
k.hand_coded_optimizations()
kernels.append(k)
if not SCHEDULE_ONLY:
asts = {x.ast.key:x.ast for x in sched if x.ast.op is UOps.SINK}.values()
kernels = []
with Profiling(PROFILE):
with Timing("***** model uops in "):
for ast in asts:
k = Kernel(ast)
k.hand_coded_optimizations()
kernels.append(k)
with Profiling(PROFILE, fn="/tmp/schedule.prof"):
with Timing("***** model linearize in "):
for k in kernels: k.linearize()
with Profiling(PROFILE, fn="/tmp/schedule.prof"):
with Timing("***** model linearize in "):
for k in kernels: k.linearize()
#renderer = Device[Device.DEFAULT].renderer
#with Profiling(PROFILE, fn="/tmp/schedule.prof"):