tinygrad/extra
George Hotz 871df1436a
more beautiful cifar (#10551)
* enumerate cases of Tensors in the JIT

* optional fused optimizers

* add fused optimizer test

* move that there

* ugh

* work on beautiful_cifar

* speed close to hlb_cifar

* schedule to corealize all

* one line sched step

* less lines
2025-05-28 20:48:20 -07:00
..
accel move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
amdpci am_smi: mem usage (#10547) 2025-05-28 16:53:31 +03:00
assembly move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
backends move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
datasets lint mlperf model_train (#10038) 2025-04-24 16:19:44 -04:00
disassemblers/adreno qcom fix disasm (#6703) 2024-09-24 15:23:43 +08:00
dsp dsp stuff / sniff ioctls from snpe (#9490) 2025-03-20 10:38:23 +08:00
gemm move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
hcqfuzz hcqfuzz: init (#10049) 2025-04-25 23:19:21 +03:00
hip_gpu_driver MI300X support (WIP) (#9585) 2025-03-29 19:46:42 +08:00
hiprtc use comgr to compile (#3248) 2024-01-26 18:27:49 -08:00
huggingface_onnx add onnx frontend stub [pr] (#9558) 2025-03-24 12:24:34 +08:00
junk coder.py can write and run code (#2439) 2023-11-25 12:27:54 -08:00
mmapeak mmapeak implementation for 7900 XTX (#10417) 2025-05-23 16:26:12 -07:00
models hotfix: that repeat_kv belongs outside the if 2025-05-11 18:43:01 -07:00
nv_gpu_driver nv: blackwell support (#10487) 2025-05-24 18:23:53 +03:00
optimization move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
profiler move viz /prof to extra [pr] (#10401) 2025-05-18 23:25:59 +03:00
qcom_gpu_driver qcom match texture/sampler descriptors to OpenCL (#7622) 2024-11-11 21:56:51 +03:00
remu add v_rcp_f32_e64 to remu (#10393) 2025-05-18 17:08:21 +03:00
resnet18 beat mlx at resnet 18 (#6611) 2024-09-20 11:28:01 +08:00
sched move fuzz_schedule.py to extra [pr] (#10444) 2025-05-21 10:07:24 +03:00
sqtt use tuple in isinstance for type checking (#9583) 2025-03-26 19:36:48 +08:00
torch_backend Add Tensor.unfold (#10518) 2025-05-26 11:15:44 -04:00
torch_hook use tuple in isinstance for type checking (#9583) 2025-03-26 19:36:48 +08:00
usbgpu usbgpu: check hash in patcher (#10266) 2025-05-12 21:08:53 +03:00
webgpu Autogen webgpu dawn, removing wgpu-py dependency (f16 support part 1) (#8646) 2025-02-07 15:16:59 +08:00
archprobe.py move dtypes to dtype.py (#2964) 2024-01-01 14:58:48 -08:00
augment.py [ready] Replacing os with pathlib (#1708) 2023-08-30 10:41:08 -07:00
bench_log.py hotfix: BenchEvent MLPERF_RUN is mlperf_run (#10526) 2025-05-26 20:19:37 -04:00
disk_read_speed.py io_uring for copies from disk (#5035) 2024-06-21 11:36:51 +03:00
dump_cache.py wow how did i think that was okay (#2339) 2023-11-16 21:21:11 -08:00
export_model.py move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
f16_decompress.py u32 to f16 in tinygrad (#8074) 2024-12-06 12:00:13 +01:00
gradcheck.py tests from grad uop path [pr] (#8313) 2024-12-18 09:25:05 -08:00
hip_events.py move autogen to runtime/autogen (#3254) 2024-01-26 12:44:19 -08:00
hip_large_kernel.py minimum change for rdna4 [pr] (#9455) 2025-03-16 13:39:24 +08:00
hook_cuda.py cuda hooking (#9180) 2025-02-20 19:20:01 +08:00
introspection.py move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
lr_scheduler.py more beautiful cifar (#10551) 2025-05-28 20:48:20 -07:00
mcts_search.py [TIP-9] rename Opt's amt to arg 2 (#8770) 2025-01-27 14:19:04 -05:00
multitensor.py multitensor start (#2676) 2023-12-07 17:07:05 -08:00
onnx.py move test over (#10508) 2025-05-25 21:51:51 -04:00
onnx_helpers.py Revert "ONNX add output shape validation (#9720)" (#9904) 2025-04-16 03:15:56 -04:00
reduce_speed.py VALIDATE_WITH_CPU [pr] (#9488) 2025-03-18 15:15:04 +08:00
replay_pkl.py hand_coded_optimizations returns list[Opt] [pr] (#9938) 2025-04-19 20:26:59 -04:00
ring_copy.py ring copy example (#3185) 2024-01-19 23:34:30 -05:00
setup_mock_amd_osx.sh add rocm 6.4 support (#10491) 2025-05-23 16:20:54 -07:00
setup_mock_nv_osx.sh hotfix: setup_mock_nv_osx 2025-02-13 12:26:15 +08:00
thneed.py new style device (#2530) 2023-11-30 17:07:16 -08:00
threefry.py feat: make buffer (#6745) 2024-09-25 18:31:03 +08:00
to_movement_ops.py fix: handle buffer size calculation in to_movement_ops and add scalar assignment test in torch_backend (#10464) 2025-05-22 10:54:13 -07:00
training.py tinytqdm.set_description and tinytrange (#5101) 2024-06-22 14:45:06 -04:00
transfer_speed.py hotfix: copy size is in bytes 2024-01-17 16:44:15 +00:00