6 Commits

Author	SHA1	Message	Date
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