github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity
Files
dcd50baca410be1389836ea781eeedb68baca551
tinygrad/test/external
History
George Hotz c5bd28e21d start work on schedule cache (#13529) 
* start work on schedule cache

* local unique

* schedule cache works

* schedule cache cleanup

* fix tests

* preserve metadata

* oops, fix cache

* put that there

* fix spec

* always miss

* why is that broken?

* src[0].op

* fix process replay

* delete abstractions2

* reenable the actual schedule cache

* metadata is best effort

* fix JIT in examples/gradaccum_mnist.py

* full jit

* fixed and test is real
2025-12-04 17:24:49 -08:00
..
mlperf_bert
delete SAVE_SCHEDULE=1 [pr] (#7087)
2024-10-16 07:13:20 +03:00
mlperf_resnet
remove tfa dependency: use keras.optimizers.Lamb and tf.raw_ops for LARS (#13555)
2025-12-03 17:48:27 -05:00
mlperf_retinanet
RetinaNet losses (#9536)
2025-03-21 15:52:54 -04:00
mlperf_stable_diffusion
Training loop for Stable Diffusion mlperf (#12315)
2025-10-03 02:45:38 -04:00
mlperf_unet3d
[MLPerf] UNet3D dataloader (#4343)
2024-04-28 22:34:18 -04:00
openpilot
fix failed threefry (#10646)
2025-06-05 17:17:42 -07:00
process_replay
require renderer argument in get_program, removes device opening in process replay [pr] (#13524)
2025-12-03 02:05:31 +08:00
sglang_llama
refactor test/external/external_llama_eval.py (#10567)
2025-06-23 10:43:20 -07:00
external_benchmark_am.py
clean: becnhmark -> benchmark (#10620)
2025-06-03 19:28:18 -07:00
external_benchmark_bert_matmuls.py
ShapeTracker.real_strides -> is_expanded [pr] (#12579)
2025-10-09 22:52:45 -04:00
external_benchmark_bert_softmax.py
delete DONT_REALIZE_EXPAND and DONT_GROUP_REDUCES (#12744)
2025-10-16 14:11:33 -04:00
external_benchmark_disk_raw.py
ci: add h machines (#11416)
2025-07-29 19:21:51 +03:00
external_benchmark_hcopt.py
move opt under codegen (#11569)
2025-08-07 14:19:17 -07:00
external_benchmark_hip_compile.py
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04:00
external_benchmark_keccak.py
Tensor.keccak("sha3_256") (#7186)
2025-06-06 15:24:05 -07:00
external_benchmark_kernel_launch.py
cleanup tests, bump caches (#11746)
2025-08-19 21:21:07 -07:00
external_benchmark_llama_schedule.py
bump line count to 14500
2025-06-23 15:32:27 -07:00
external_benchmark_load_stable_diffusion.py
gate METAL_FAST_LOAD
2023-12-01 15:28:40 -08:00
external_benchmark_multitensor_allreduce.py
add some options to allreduce bench (#9348)
2025-03-04 23:46:36 +03:00
external_benchmark_op_conv.py
one last regressed dm kernel (#13201)
2025-11-10 23:30:52 -05:00
external_benchmark_pyrender.py
add external_benchmark_pyrender (#13378)
2025-11-20 17:38:28 +08:00
external_benchmark_resnet.py
delete SAVE_SCHEDULE=1 [pr] (#7087)
2024-10-16 07:13:20 +03:00
external_benchmark_schedule.py
control_flow.py -> linearizer.py [pr] (#12948)
2025-10-27 12:38:13 -04:00
external_benchmark_sdxl_softmax.py
update Kernel API in tests + move optimize_local_size (#11907)
2025-08-28 15:12:47 -07:00
external_cl_half_max.py
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04:00
external_fuzz_ampt.py
Fix/hcqfuzz harnesss bug (#10923)
2025-06-23 11:22:30 +03:00
external_fuzz_hcq_mp.py
system: fix flock on pcidevs (#13123)
2025-11-06 19:02:13 +08:00
external_fuzz_hcq_signals.py
hcqfuzz: init (#10049)
2025-04-25 23:19:21 +03:00
external_fuzz_tlsf.py
Fix/hcqfuzz harnesss bug (#10923)
2025-06-23 11:22:30 +03:00
external_gpu_fail_osx.py
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04: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_llm_eval.py
BPE tokenizer (#11415)
2025-08-04 09:52:38 -07:00
external_metal_compile_fail.py
Use MTLCompiler directly (v2) (#7920)
2024-12-04 16:36:48 +08:00
external_metal_compile_slow.py
add Scheduler to replace Kernel with POSTOPT=2 (#11924)
2025-09-03 19:23:30 -07:00
external_model_benchmark.py
move frontend dir to nn [pr] (#12470)
2025-10-07 10:42:22 +08:00
external_multi_gpu.py
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04:00
external_nhwc_conv.py
add fun with nhwc convs
2025-10-28 17:12:22 +08:00
external_openpilot_image_warp.py
openpilot warp (#13283)
2025-11-14 13:55:32 -08:00
external_osx_profiling.py
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04:00
external_test_am.py
allocate less memory in am tests (#11342)
2025-07-23 11:11:26 +03:00
external_test_amd.py
rename lazydata to uop (#10698)
2025-06-08 08:42:22 -07:00
external_test_datasets.py
skip test TestKiTS19Dataset::test_training_set (#10936)
2025-06-23 14:27:24 -04:00
external_test_dev_var.py
fix up some slow tests that launch python (#12672)
2025-10-14 19:13:55 +08:00
external_test_embedding.py
make embedding and GPT-2 fast (#1631)
2023-08-22 15:14:38 -07:00
external_test_example.py
print special ops in postrange (#13318)
2025-11-17 14:43:23 -08:00
external_test_hcq_fuzz_failures.py
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04:00
external_test_hcq_open.py
test hcq open with pytest (#13124)
2025-11-06 20:09:51 +08:00
external_test_hcq.py
rename lazydata to uop (#10698)
2025-06-08 08:42:22 -07:00
external_test_hip_compile.py
move opt under codegen (#11569)
2025-08-07 14:19:17 -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
ops_gpu -> ops_cl (#12103)
2025-09-10 15:15:48 -04: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_keccak.py
Tensor.keccak("sha3_256") (#7186)
2025-06-06 15:24:05 -07:00
external_test_llama3_ff.py
work to make GEMV fast (#5824)
2024-07-30 17:41:40 -07: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
RetinaNet losses (#9536)
2025-03-21 15:52:54 -04:00
external_test_mamba.py
external that test
2024-03-29 19:35:50 -07:00
external_test_metrics.py
log_perplexity metrics (#10912)
2025-06-21 10:44:47 -04: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
move opt under codegen (#11569)
2025-08-07 14:19:17 -07:00
external_test_onnx_backend.py
add max/min reduction support to ScatterND (#13562)
2025-12-04 00:53:47 -08:00
external_test_onnx_ops.py
Onnx Layer/Group/RMS/Batch-Norm ReduceL2 fp32 intermediates for fp16 (#12109)
2025-10-24 12:26:11 +02:00
external_test_onnx_runner.py
move frontend dir to nn [pr] (#12470)
2025-10-07 10:42:22 +08:00
external_test_opt.py
remove trivial use of RANGEIFY flag (#12550)
2025-10-09 02:29:38 -04:00
external_test_optim.py
remove tfa dependency: use keras.optimizers.Lamb and tf.raw_ops for LARS (#13555)
2025-12-03 17:48:27 -05:00
external_test_process_count.py
add external_test_process_count
2025-12-02 17:26:30 -08:00
external_test_simple_tokenizer.py
clean up the LLM tokenizer (#12653)
2025-10-14 14:22:01 +08:00
external_test_speed_llama.py
make method_cache account for compiler (#12156)
2025-09-13 17:00:11 +03:00
external_test_speed_theoretical.py
test flops (and allow wide ALU in UOps) [run_process_replay] (#5749)
2024-07-26 21:07:28 -07:00
external_test_tlsf.py
factor out generic MemoryManager (#10910)
2025-06-21 16:18:33 +03:00
external_test_usb_asm24.py
usbgpu: 10 steps in benchmark to hit cache (#10273)
2025-05-13 17:06:50 +03: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
More yolo processing in tinygrad (#9928)
2025-04-24 16:21:46 -04:00
external_uop_gc.py
start work on schedule cache (#13529)
2025-12-04 17:24:49 -08:00
fuzz_fast_idiv.py
clean up the spec (#12868)
2025-10-22 19:50:42 +08:00
fuzz_graph.py
rename lazydata to uop (#10698)
2025-06-08 08:42:22 -07:00
fuzz_kfd.py
unify to HWQueue [pr] (#7812)
2024-11-21 10:33:08 +08:00
fuzz_linearizer.py
rename opts to ren, add store ranges back (#12856)
2025-10-22 09:15:38 +08:00
fuzz_shape_ops.py
move fuzz_shape_ops to run with other fuzzer (#10767)
2025-06-10 17:43:04 -04:00
fuzz_shapetracker_math.py
remove ShapeTracker.invert (#12520)
2025-10-08 18:37:34 +08:00
fuzz_shapetracker.py
[bounty] Stride is flip (#8792)
2025-01-31 11:34:10 +09:00
fuzz_symbolic_symbolic_div.py
Symbolic divisor fuzzer (#13433)
2025-11-23 20:29:32 +01:00
fuzz_symbolic.py
clean up the spec (#12868)
2025-10-22 19:50:42 +08:00
graph_batchnorm.py
with Tensor.train() (#1935)
2023-09-28 18:02:31 -07:00
speed_compare_amd_am.py
move opt under codegen (#11569)
2025-08-07 14:19:17 -07:00
speed_compare_cuda_nv.py
move opt under codegen (#11569)
2025-08-07 14:19:17 -07:00
speed_compare_cuda_ptx.py
move opt under codegen (#11569)
2025-08-07 14:19:17 -07:00
speed_v_theoretical.py
lower nv test_gemm_4096 target (#13107)
2025-11-05 11:05:16 -05:00
verify_kernel.py
move opt under codegen (#11569)
2025-08-07 14:19:17 -07:00