George Hotz dc21e63bd2 test: put conv in one reduce (#4441) ... * test: put conv in one reduce * put reduce at the end * more expand * generic, and that expand was breaking things * ratio * don't undo the expand * arg 1 * strides * warning, for resnet * warning removed * disable cast * handle cast * op * err, that's right * fixup * fix that * a test to play with * add double_reduces * working up to final reshape * fold the last reshape * moved to schedule * fix axis * ci, need to bring arange back * FUSE_CONV_BW maybe * valid in 3.9 * test_expand_reduce_is_folded_on_different_axes * add FUSE_CONV_BW=1 * test_fold_batchnorm_backward * test_sgd_4convs_fuse --------- Co-authored-by: qazal <qazal.software@gmail.com>		2024-07-22 12:16:13 +03:00
..
mlperf_bert	Add layerwise performance bench for bert (#5349)	2024-07-09 15:03:25 -04:00
mlperf_resnet	PolynomialDecayWithWarmup + tests (#3649)	2024-03-07 18:53:36 -05:00
mlperf_unet3d	[MLPerf] UNet3D dataloader (#4343)	2024-04-28 22:34:18 -04:00
process_replay	hotfix: don't assert process replay in master. (#5562)	2024-07-18 22:05:00 +03:00
external_benchmark_hip_compile.py	hip compile speed (#2606)	2023-12-04 13:47:40 -08:00
external_benchmark_load_stable_diffusion.py	gate METAL_FAST_LOAD	2023-12-01 15:28:40 -08:00
external_benchmark_multitensor_allreduce.py	move image into tensor.py. delete features (#4603)	2024-05-15 10:50:25 -07:00
external_benchmark_resnet.py	ruff: unnecessary-comprehension (#5174)	2024-06-27 07:45:29 -04:00
external_benchmark_schedule.py	MetaOps.KERNEL (#5543)	2024-07-17 19:41:23 -07:00
external_cl_half_max.py	use default dict for external_model_benchmark (#2592)	2023-12-03 15:25:43 -08:00
external_hip_compiler_bug.py	CompiledASTRunner -> CompiledRunner (#4148)	2024-04-11 08:49:52 -07:00
external_jit_failure.py	fix jit realize issue (#3258)	2024-01-26 18:27:35 -08:00
external_llama_eval.py	ruff checks the max line length is 150 (#2734)	2023-12-12 17:34:47 -08:00
external_metal_compile_fail.py	metal compile fail	2024-07-11 19:27:05 -07:00
external_model_benchmark.py	ruff: unnecessary-comprehension (#5174)	2024-06-27 07:45:29 -04:00
external_multi_gpu.py	move disassemblers and openpilot (#4592)	2024-05-14 19:30:02 -07:00
external_osx_profiling.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08:00
external_slow_global_dim4_resnet.py	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07:00
external_test_amd.py	amd doorbell size is 64bits (#4448)	2024-05-06 16:59:59 +03:00
external_test_datasets.py	UNet3D dataloader shared memory fix (#5465)	2024-07-13 20:26:00 -04:00
external_test_embedding.py	make embedding and GPT-2 fast (#1631)	2023-08-22 15:14:38 -07:00
external_test_example.py	numpy device + pickle it (#4120)	2024-04-09 13:19:30 -07:00
external_test_hcq.py	add _alloc_signal/_free_signal to hcq (#5264)	2024-07-02 23:35:39 +03:00
external_test_hip_compile.py	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07:00
external_test_hsa_driver.py	Rename tinygrad/runtime/driver to support (#5413)	2024-07-12 11:06:42 -07:00
external_test_image.py	fixes (#1893)	2023-09-22 07:20:27 +08:00
external_test_jit_on_models.py	Pulled CLIP and UNet into Seperate Files (#5253)	2024-07-01 22:33:01 -04:00
external_test_lm_head.py	isolate the 134ms kernel in train_gpt2.py (#4773)	2024-05-29 17:26:24 -04:00
external_test_losses.py	[MLPerf][UNet3D] Add DICE loss + metrics (#4204)	2024-04-17 20:09:33 -04:00
external_test_mamba.py	external that test	2024-03-29 19:35:50 -07:00
external_test_metrics.py	Convert BinaryOps.DIV to UnaryOps.RECIP and BinaryOps.IDIV (#4887)	2024-06-14 02:43:46 -07:00
external_test_mnist_data_select.py	add quick external data select test	2024-03-02 05:38:32 -08:00
external_test_nv.py	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07:00
external_test_onnx_backend.py	remove "no-nans-fp-math"="true" for LLVM (#5282)	2024-07-03 17:52:50 -04:00
external_test_opt.py	test: put conv in one reduce (#4441)	2024-07-22 12:16:13 +03:00
external_test_optim.py	improve test_dropout_on_shard (#4912)	2024-06-11 11:36:02 -04:00
external_test_speed_llama.py	all realize 2 (#4527)	2024-05-10 22:43:09 -07:00
external_test_uops_graphing.py	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07:00
external_test_whisper_librispeech.py	names shadowing builtins (#5179)	2024-06-27 08:15:01 -04:00
external_test_yolo.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
external_test_yolov8.py	ruff checks the max line length is 150 (#2734)	2023-12-12 17:34:47 -08:00
fuzz_graph.py	graph fuzzer (#5082)	2024-06-21 18:47:23 +03:00
fuzz_kfd.py	add _alloc_signal/_free_signal to hcq (#5264)	2024-07-02 23:35:39 +03:00
fuzz_linearizer.py	pretty print lazy op per default (#5505)	2024-07-18 09:34:08 -07:00
fuzz_schedule.py	s/loadops/metaops [run_process_replay] (#5421)	2024-07-12 13:26:50 -07:00
fuzz_shapetracker.py	minor improvements (#2845)	2023-12-18 22:09:08 -08:00
fuzz_shapetracker_math.py	tinytqdm.set_description and tinytrange (#5101)	2024-06-22 14:45:06 -04:00
fuzz_symbolic.py	Variable.num -> NumNode (#2354)	2023-11-18 15:45:52 -05:00
fuzz_uops.py	move uopgraph to file [run_process_replay] (#5364)	2024-07-10 17:34:50 -07:00
graph_batchnorm.py	with Tensor.train() (#1935)	2023-09-28 18:02:31 -07:00
speed_beam_v_hcopt.py	move graph/search to engine (#4596)	2024-05-14 23:12:59 -07:00
speed_compare_cuda_nv.py	move colorize_float to helpers.py (#5490)	2024-07-15 11:29:03 -07:00
speed_compare_cuda_ptx.py	move colorize_float to helpers.py (#5490)	2024-07-15 11:29:03 -07:00
verify_kernel.py	pretty print lazy op per default (#5505)	2024-07-18 09:34:08 -07:00