diff --git a/test/external/external_benchmark_schedule.py b/test/external/external_benchmark_schedule.py new file mode 100644 index 0000000000..991d051d66 --- /dev/null +++ b/test/external/external_benchmark_schedule.py @@ -0,0 +1,23 @@ +from extra.models.resnet import ResNet50 +from tinygrad import Tensor +from tinygrad.helpers import Profiling, Timing, getenv +from tinygrad.engine.realize import lower_schedule + +if __name__ == "__main__": + mdl = ResNet50() + img = Tensor.empty(64, 3, 224, 224) + + PROFILE = getenv("PROFILE", 1) + + with Profiling(PROFILE): + with Timing("***** model forward in "): + out = mdl(img) + + with Profiling(PROFILE): + with Timing("***** model schedule in "): + sched = out.schedule() + + with Profiling(PROFILE): + with Timing("***** model lower in "): + ei = list(lower_schedule(sched)) +