geohotstan 8c0d0a122c Add return_indices to max_pool (#9506) ...

* wow argmax is so good

* 1 less line

* clean up and better variable names

* is this torch thing right...?

* add more tests

* slap a TODO on it

* clean ups

* prettier looking code and fix ceil mode test

* add return types and some docs

* ok that was a bad example since indices == value, just no example

2025-03-19 15:25:37 -04:00

..

accel

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

amdpci

am: rename soc21 to soc (#9482)

2025-03-18 08:54:26 +08:00

assembly

s/UOps/Ops (#7500)

2024-11-03 11:26:10 +08:00

backends

CLANG -> CPU (#9189)

2025-02-20 18:03:09 -05:00

datasets

do not construct unmasked VALID (#8759)

2025-01-28 20:51:21 +02:00

disassemblers/adreno

qcom fix disasm (#6703)

2024-09-24 15:23:43 +08:00

dsp

dsp simulator (#8869)

2025-02-04 09:45:04 +08:00

gemm

extra/gemm/max_matmul: start of custom kernels for GEMM (#6926)

2025-03-19 15:04:57 +08:00

hip_gpu_driver

amd: autogen ip bases (#9360)

2025-03-05 22:30:38 +03:00

hiprtc

use comgr to compile (#3248)

2024-01-26 18:27:49 -08:00

huggingface_onnx

benchmark huggingface onnx models (#8493)

2025-03-12 20:13:12 -04:00

junk

coder.py can write and run code (#2439)

2023-11-25 12:27:54 -08:00

models

olmoe (from stream, wip) (#9390)

2025-03-10 13:46:33 +08:00

nv_gpu_driver

nv fix shared_memory_size (#7239)

2024-10-23 21:59:47 +03:00

optimization

fix import time_linearizer [pr] (#9118)

2025-02-15 21:33:28 -05:00

qcom_gpu_driver

qcom match texture/sampler descriptors to OpenCL (#7622)

2024-11-11 21:56:51 +03:00

resnet18

beat mlx at resnet 18 (#6611)

2024-09-20 11:28:01 +08:00

sqtt

SQTT profiling (#9278)

2025-03-11 13:19:56 +08:00

torch_backend

Add return_indices to max_pool (#9506)

2025-03-19 15:25:37 -04:00

torch_hook

torch_hook fixes (#9334)

2025-03-03 23:07:30 +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

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

tinychat in browser, Part 2: model export (#9274)

2025-03-04 15:53:30 +08: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

rename LazyBuffer -> UOp [pr] (#8169)

2024-12-11 16:15:52 -08:00

lr_scheduler.py

use at least float32 for optim.lr (#4297)

2024-04-25 14:42:28 -04: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_helpers.py

benchmark huggingface onnx models (#8493)

2025-03-12 20:13:12 -04:00

onnx.py

Add return_indices to max_pool (#9506)

2025-03-19 15:25:37 -04:00

reduce_speed.py

VALIDATE_WITH_CPU [pr] (#9488)

2025-03-18 15:15:04 +08:00

replay_pkl.py

dsp work try 3 (#9475)

2025-03-17 16:42:12 +08:00

ring_copy.py

ring copy example (#3185)

2024-01-19 23:34:30 -05:00

setup_mock_amd_osx.sh

add script to install amd mockgpu on macOS (#8536)

2025-01-09 01:29:25 +03: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

full fix for as_strided in torch backend (#9257)

2025-02-26 22:34:05 +08: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