10 commits

Author	SHA1	Message	Date
chenyu	8631b6f17d	remove use of requires_grad in test/ (#16237)	2026-05-16 17:21:07 -04:00
nimlgen	01ac1c8c15	remove all run_schedule from tests (#15846)	2026-04-21 12:02:10 +03:00
qazal	390171d686	delete SAVE_SCHEDULE=1 [pr] (#7087)	2024-10-16 07:13:20 +03:00
Roelof van Dijk	f88f71d73a	ruff: unnecessary-comprehension (#5174) ... * enable ruff C416 unnecessary-comprehension * already a list	2024-06-27 07:45:29 -04:00
chenyu	24a6342950	add mem/s to external_benchmark_resnet (#4309)	2024-04-26 20:07:17 -04:00
David Hou	6f792b727b	More improvements for resnet layer bench (#4272) ... * fix first layer size, new schedule stuff * estimates * get different conv layers * \r for estimated times * E501 * space after comma	2024-04-25 12:40:49 -04:00
George Hotz	967638f0d5	update docs, remove corealize (#4264) ... * update docs, remove corealize * handle 0 line count * tensor schedule	2024-04-23 12:05:29 +04:00
David Hou	dc4b1af09c	more realistic edge behavior for resnet benchmark (#4231) ... * more realistic edge behavior for resnet benchmark * schedule_step * realize all parameters ahead of time * don't save setup and misc schedules	2024-04-19 20:07:46 -04:00
David Hou	7fb220a567	touchup resnet_layer_bench (#4191)	2024-04-16 14:43:00 -04:00
David Hou	1dbf3b2b19	Benchmarks for individual resnet layers (#4182) ... * resnet individual layer benchmarks! * small * 1 and 2 * mem_used * no ci * better conv print * defaults * prints * adjust * adjust * adjust * benchmark only one layer example * tensor.training, zero_grad, sum instead of mean, last mem, last kernel count * default jitcnt=1 * scale flops/kernels with jitcnt * add note about jitcnt memory * touchup	2024-04-16 13:53:18 -04:00