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
11,856 Commits 248 Branches 14 Tags
bf2d9d138ff47cd8965f84652e7bfa0a086c4dfd
Commit Graph

995 Commits

Author SHA1 Message Date
chenyu
cb69b7b2b2 comment out fold_where_closure (#14316) 2026-01-24 10:15:42 -05:00
chenyu
e65bc7a7c5 where closure folding (#14304) 2026-01-23 10:55:13 -05:00
chenyu
5f32f7a06b fix winograd padding order (#14294) 2026-01-22 23:00:14 -05:00
chenyu
6279ae4a94 remove llm generate always reset start_pos (#14276) 
* remove llm generate always reset start_pos

by itself seems like a bug, also added a test to repro forward_jit.reset() issue

* issue is jit graph, so revert that test
 2026-01-21 16:54:30 -05:00
chenyu
e64111ad08 update all_same [pr] (#14270) 
add type annotation and unit test
 2026-01-21 11:26:15 -05:00
George Hotz
5e24643889 minor import speedups (#14244) 
* minor import speedups

* server stuff in server places

* pre-commit

* fix
 2026-01-20 15:05:36 +09:00
qazal
b1c5a242b7 Revert "move is_dtype_supported logic to renderer (#14188)" (#14237) 
This reverts commit 161fee9a48.
 2026-01-20 12:19:14 +09:00
Christopher Milan
161fee9a48 move is_dtype_supported logic to renderer (#14188) 
* move is_dtype_supported logic to renderer

* fix CPU_COUNT

* mypy happy

* early import libclang too with llvm

* run with debug

* skip autogen tests if MTLCompiler or llvm is loaded

* run autogen tests separately in CI

* lint
 2026-01-18 22:37:04 -05:00
chenyu
e7c2df9113 improve consecutive Tensor indexing (#14208) 
* improve consecutive Tensor indexing

instead of O(idx_counts*src_dims), it can just be O(idx_counts)

* test correctness
 2026-01-18 15:14:33 -05:00
chenyu
c7b8f6496f remove dtypes.index_like and dtypes.fields [pr] (#14207) 
barely used, so just use inline and DTYPES_DICT
 2026-01-18 11:49:01 -05:00
Christopher Milan
a021b84604 autogen: fix enum (#14171) 2026-01-16 01:30:11 -05:00
chenyu
14e9a71a41 move test_assign to unit (#14165) 
scheduling these should not depend on device
 2026-01-15 17:10:13 -05:00
Christopher Milan
0cb024a5bb remove ctypes.Structure (#13651) 2026-01-15 05:06:22 -05:00
qazal
164bc678a6 scheduler: sched_cache bugfix for different Tensor.custom_kernel schedules (#14161) 
* simplest failing test

* min fix

* same function reuses the cache

* SPEC=2 never worked for custom_kernel
 2026-01-15 14:59:14 +09:00
chenyu
35c9701df0 update outdated tests and comments (#14090) 2026-01-10 01:00:48 -05:00
chenyu
92246ea731 update tests, WEBGPU=1 pytest . passes (#14089) 
* update tests, `WEBGPU=1 pytest .` passes

* minor update
 2026-01-10 00:03:02 -05:00
chenyu
eacccc5ace more disk assign tests (#14087) 
covers more edge cases
 2026-01-09 14:14:52 -05:00
chenyu
ed295e74dc don't skip gguf test if ggml is not installed (#14086) 
* don't skip gguf test if ggml is not installed

should just let it fail

* fix
 2026-01-09 12:05:58 -05:00
chenyu
cff33c8d78 add some disk assign tests (#14085) 2026-01-09 11:50:59 -05:00
Garret Castro
16b652302e skip bf16 test if not supported by device (#14070) 2026-01-08 13:37:24 -05:00
Christopher Milan
b2a0b9c551 autogen: dump patch in CI (#14010) 
* autogen: don't fast-fail, produce patch artifact on differences

All verification steps now use continue-on-error to run completely.
Each job generates a patch artifact containing all differences found.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

* add gen from header test

* fix tests

* fail if diff

* add forward decl autogen test

* remove confusing/wrong comments

* macos unittests set LIBCLANG_PATH

---------

Co-authored-by: Claude Sonnet 4.5 <noreply@anthropic.com>
 2026-01-04 22:38:12 -05:00
chenyu
cfb8bf5814 faster image load (#13977) 
sometimes image load does not need to init with NAN
 2026-01-04 13:09:59 -05:00
chenyu
8003db2a28 test case of NOOP store load folding (#13997) 2026-01-03 14:39:26 -05:00
chenyu
2e2b5fed12 fix misspellings (#13976) 2026-01-02 10:37:38 -05:00
chenyu
af0392efea only set DiskDevice.size if it opens successfully (#13962) 2026-01-01 19:33:26 -05:00
chenyu
e036d6df89 properly fix DiskDevice reuse (#13961) 2026-01-01 18:08:23 -05:00
chenyu
cb7c76a3bd update test_fuzz_failure to not contruct full UOp (#13960) 2026-01-01 15:09:58 -05:00
chenyu
8e416df438 simpler InvalidType [pr] (#13957) 
simpler singleton pattern
 2026-01-01 13:55:51 -05:00
chenyu
4d5c4d256d update tqdm for edge case (#13956) 
1.00kit/s and not 1000it/s for value 999.5
 2026-01-01 11:37:26 -05:00
chenyu
b91b46091c delete test_tensor_uop (#13951) 
old test for shape tracker. also update tests that refer shapetracker

names
 2026-01-01 09:25:05 -05:00
chenyu
17ef4af72c new ceildiv that fixed symbolic conv (#13944) 
* new ceildiv that fixed symbolic conv

* smaller test case
 2026-01-01 09:02:41 -05:00
haofei
526fd4ec71 Fix SVD rank‑1 Jacobi rotation when tau == 0 (#13945) 2026-01-01 00:30:18 -05:00
haofei
20777f30b9 Fix QR/SVD NaNs on zero/orthogonal inputs (#13943) 2025-12-31 23:40:09 -05:00
chenyu
52acadc160 consolidate IGNORE_OOB=0 tests (#13937) 
add a new unit test file and add more cases
 2025-12-31 15:24:20 -05:00
chenyu
0a98fd38b3 fix tests that failed locally on mac (#13872) 
keccak output was silently broken without contiguous
 2025-12-29 11:23:38 -05:00
Clément Verrier
0e409ff5ce fix indentation in UOp pretty_print for repeated references (#13857) 
* fix correct indentation in UOp pretty_print for repeated references

When a UOp was referenced multiple times, the walrus operator notation
(e.g., x0:=) was correctly used for the first occurrence, but subsequent
references had misaligned indentation due to an extra space character.

Fix indentation misalignment in pretty_print() when UOps are referenced
multiple times.

* add simple unit tests for UOp repr

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-12-29 10:46:16 -05:00
anu
9b4de8abc7 fix beam in python 3.14+ (#13836) 
* fix beam search on python 3.14

* add PickleableCount class to helpers

* change name, add test, add step

* tidy count init
 2025-12-27 16:24:22 -05:00
chenyu
54af29dbdb trange can just be a function (#13827) 2025-12-24 23:57:10 -05:00
George Hotz
43c6e973d8 add optional compiler in Renderer (#13817) 
* add optional compiler in Renderer [pr]

* fix

* late init

* remove precompiled

* cleanup
 2025-12-23 17:58:46 -05:00
George Hotz
6439a515be test fixups / speedups / var_vals refactor (#13812) 
* no PYTHONPATH + llm server port 0

* llm tok speedup

* refactor var_vals
 2025-12-23 12:05:59 -05:00
George Hotz
8dcba2e2cc no full_rewrite [pr] (#13809) 
* no full_rewrite [pr]

* fix

* fix docs
 2025-12-22 23:20:01 -05:00
George Hotz
df0f9d6860 add olmoe support to llm (#13792) 
* add olmoe support to llm

* cleanups

* simpler

* clean

* fix mypy

* lil

* remove dumb assert
 2025-12-22 10:41:35 -04:00
chenyu
5cb827f7bf clean up can_lossless_cast and add missing pairs [p] (#13793) 2025-12-21 12:18:33 -05:00
George Hotz
75a6a03664 add qwen3 moe support to tinygrad.apps.llm (#13775) 
* qwen moe works

* simple moe

* one test

* integration
 2025-12-21 12:36:02 -04:00
chenyu
733ef0452c update test_uop_resolve (#13777) 
plain @unittest.expectedFailure is too broad
 2025-12-20 12:40:59 -05:00
chenyu
185a000882 gradient of COPY (#13760) 2025-12-19 13:33:59 -05:00
George Hotz
aeb7516c8a tests passing on tinybox h3 (#13742) 2025-12-17 19:04:34 -04:00
George Hotz
b013244c38 fix local tests for AMD_LLVM (#13738) 
* fix local tests for AMD_LLVM

* fix linters

* skip that for now

* fix segfault
 2025-12-17 12:23:46 -04:00
George Hotz
3dbde178c1 mark slow tests as slow instead of as CI (#13736) 
* mark slow tests as slow instead of as CI

* CI shouldn't have different behavior

* more skips / CI

* slow
 2025-12-17 10:29:57 -04:00
George Hotz
9015a22523 make tests faster (#13734) 2025-12-17 09:39:44 -04:00