tinygrad/extra
Sieds Lykles cc038b31b6
Shrink instead of reshape to unregister symbolic (#12241)
* Slice to unbind symbolic

* use vmax for now

* assert shape in reshape is valid

* update test_symbolic_ops to use shrink instead of reshape

* remove infer_with_bound_values for npw

* symbolic output doesnt have symbolic strides

* symbolic jit tests use shrink to unregister symbolic

* update test

* update more tests

* wrap vmax in int()

* only create a new st if the store is not an assigne

* unwrap st

* comments
2025-09-19 06:04:35 +02:00
..
amdpci am: init support for aql (#11888) 2025-08-28 18:41:46 +03:00
assembly ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
backends var_vals uses str for var (#12011) 2025-09-06 04:16:12 +02:00
datasets very tiny generate_dataset (#11013) 2025-06-27 17:10:45 -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 ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
hcq ast seems to probe nv as well (#11494) 2025-08-04 11:47:07 +03:00
hcqfuzz remove FUSE_ARANGE_UINT (#11567) 2025-08-07 16:49:06 -04:00
hip_gpu_driver hip_ioctl: platform.machine added (#11084) 2025-07-04 17:20:24 +03:00
hiprtc use comgr to compile (#3248) 2024-01-26 18:27:49 -08:00
huggingface_onnx Add Onnx Huggingface to test/models/test_onnx.py (#11468) 2025-08-14 11:16:41 -04: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 don't hardcode weights path (#12171) 2025-09-15 00:33:47 -04:00
nv_gpu_driver auto-select available compilers (#12094) 2025-09-10 19:52:01 +03:00
optimization Shrink instead of reshape to unregister symbolic (#12241) 2025-09-19 06:04:35 +02:00
perfetto upd perfetto (#11528) 2025-08-06 14:00:34 +03:00
qcom_gpu_driver ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
remu remu: add v_rsq_f32_e32 instruction (#11947) 2025-09-01 11:29:31 +03:00
resnet18 remove Tensor.no_grad, it's meaningless now [pr] (#10556) 2025-05-28 22:20:02 -07:00
sched move fuzz_schedule.py to extra [pr] (#10444) 2025-05-21 10:07:24 +03:00
sqtt remove PROFILE=1 option, it's just VIZ=1 [pr] (#12176) 2025-09-15 12:51:50 +03:00
torch_backend update torch 2.8 (#12172) 2025-09-14 15:19:03 -04:00
torch_hook rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07: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 ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04: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 ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04: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 var_vals uses str for var (#12011) 2025-09-06 04:16:12 +02:00
multitensor.py rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07:00
onnx_helpers.py onnx in tinygrad (#11675) 2025-08-14 19:57:21 -04:00
reduce_speed.py VALIDATE_WITH_CPU [pr] (#9488) 2025-03-18 15:15:04 +08:00
replay_pkl.py update Kernel API in tests + move optimize_local_size (#11907) 2025-08-28 15:12:47 -07: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
test_hcopt.py add WARP axistype, fix postopt bugs (#12033) 2025-09-05 10:36:55 -07:00
test_pyrender.py test pyrender (#12005) 2025-09-04 11:48:40 -07:00
thneed.py ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
threefry.py feat: make buffer (#6745) 2024-09-25 18:31:03 +08:00
to_movement_ops.py update torch 2.8 (#12172) 2025-09-14 15:19:03 -04:00
torch_muon.py [bounty] Muon optim (#11414) 2025-08-13 14:27:55 -04: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