sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 6 Activity
11,453 Commits 336 Branches 15 Tags
remove_programspec
Commit Graph
4777 Commits
Author SHA1 Message Date
qazalandGitHub 935cd01f56 simple failing test for graph_rewrite children [pr] (#9489) 
* simple failing test for graph_rewrite children [pr]

* lint

* update too
 2025-03-18 13:07:21 +08:00
Anish UmaleandGitHub 5e58f4b65b Tiny backend test_ops fix part 3 (#9483) 
* extract straightforward things from https://github.com/tinygrad/tinygrad/pull/9302

* pass dtype and device for ones_like
 2025-03-17 18:01:51 -04:00
TJGitHubGeorge Hotz
 		9fcef4d009 add masked_select to tensor.py (#9468) 
* add masked_select to tensor.py

* fix tests

---------

Co-authored-by: George Hotz <[email protected]>
 2025-03-17 16:05:36 -04:00
chenyuandGitHub 4f8eac59ea failed test case for threefry (#9469) 
* failed test case for threefry

not sure if it's always like this, but increment before _threefry_random_bits is incorrect. the counts should start with random numbers generated so far.

use jax to generate 20 + 20 + 10 random numbers, the first 20 + 20 matches and the last 10 are different. just moving increment after _threefry_random_bits matches the number but jit test failes

* workaround

* why is this different?

* revert those

* and that
 2025-03-17 14:52:10 -04:00
geohotstanGitHubchenyu
 		53d6f1e1bb Add bitonic cat sort (#9422) 
* poc

* repeated values fail, sigh

* is this being timed out?

* fix up down names

* bitonic v2, does this run?

* bitonic v3, faster

* bitonic v3.1, faster

* bitonic v3.1.1, same speed unlucky

* support dim and indices

* bitonic v3.2, simpler code, TODO repeated indices

* bruv gimme green for once cmon

* cat (stack) implementation, slow but maybe one day when cat is fast meow

* revert to v3.2

* bitonic v4, who let the cats out edition

* clean up variable names

* figured out repeated indices :D

* ruff check --fix

* use sort for topk

* add Tensor.sort everywhere

* fix docs and add some types

* slightly better variable names

* am I doing torch inplace correctly?

* delegate sort to values_stable

* add a contig, faster first sort

* maybe don't test_inplace

---------

Co-authored-by: chenyu <[email protected]>
 2025-03-17 12:01:23 -04:00
qazalandGitHub e03c0aacf2 more explicit DONT_PUSH_VIEWS [pr] (#9479) 
* more explicit DONT_PUSH_VIEWS [pr]

* update tests to not handcode ast

* lint

* test_recursive_swizzle and test_simple_store_reshape
 2025-03-17 20:43:21 +08:00
qazalandGitHub 3b00a778ba fix view_left for unsafe pad ops [pr] (#9478) 2025-03-17 19:02:02 +08:00
qazalandGitHub 813f713edc merge_views for buffer ops + create valids last (#9472) 
* merge_views for buffer ops + create valids last

* view.arg

* pass
 2025-03-17 17:15:44 +08:00
qazalandGitHub bd1f71c1e2 simple failing test for extra ops in VALID [pr] (#9474) 
* simple failing test for extra valids [pr]

* this has DEBUG=4
 2025-03-17 17:02:40 +08:00
qazalandGitHub e26caf4c3a hotfix: skip test_mean_half_precision_underflow on amd ci (#9476) 
The global size is very large (781250 gidx) and the emulated version takes more than 1
minute to execute the kernel.
 2025-03-17 16:47:48 +08:00
George HotzandGitHub 15ee742afa add get_children_map to uop (#9470) 
* add get_children_map to uop

* update_children

* fix new children
 2025-03-17 14:36:13 +08:00
George HotzandGitHub cb7a7f69c7 quantization preprocessor from DSP, should be universal (#9437) 
* quantization preprocessor from DSP, should be universal

* touchups

* fix tests
 2025-03-15 07:49:37 +08:00
chenyuandGitHub 99b0287e4e add GROUP and GROUPTOP to test_arange (#9432) 
it does not grow quadratically, but it's not 0 ops now
 2025-03-13 11:28:38 -04:00
qazalandGitHub 90ffa9bd45 swizzle without buffer ops try 2 [pr] (#9427) 
* add DONT_PUSH_VIEWS to matchers

* swizzle without buffer ops try 2 [pr]

* swizzle reduceop

* simple failing test

* fix failing test

* s/on/for
 2025-03-13 10:00:40 +01:00
chenyuandGitHub 22fc0a2e36 bert sum acc in half (#9412) 
also BS=96
 2025-03-11 23:03:15 -04:00
George HotzandGitHub e174c6c3bc new devectorizer (#9331) 
* new devectorizer

* lidx

* test linearizer passes

* fix images

* fix unfoldable image load

* delete unused

* improve fix_unfoldable_image_load

* working for image

* fixup types

* fixup transcendental

* cast_vec

* cleaner transcendental

* skip failing test

* err, flip that

* not devec

* sqrt
 2025-03-11 18:47:56 +08:00
George HotzandGitHub 2780e2027e devectorize prereqs [pr] (#9404) 2025-03-11 12:33:29 +08:00
chenyuandGitHub 01e8b60911 acc_dtype -> dtype (#9402) 
matched numpy and torch
 2025-03-10 16:05:30 -04:00
qazalandGitHub 59dfb234eb replace hardcoded ast with tensors in TestSwizzle [pr] (#9401) 2025-03-10 19:33:57 +01:00
geohotstanandGitHub 1d64c12f2b add Topk to tensor (#9343) 
* terrible but somewhat working impl

* linux behaves differently than macos?

* slightly better impl

* small clean up; haven't figured this out yet

* better

* torch has different behavior on linux and macos for duplicated values

* add sum docs

* fix test

* add torch return_type test

* add an exception test

* wrap_fxn instead, and move op lower in order

* better repeated values test

* rerun ci
 2025-03-09 20:01:42 -04:00
qazalandGitHub a1f41fadf6 test_schedule cleanups + add DONT_GROUP_REDUCES [pr] (#9392) 
* test_schedule cleanups + add DONT_GROUP_REDUCES [pr]

* replace with test_swizzle_reduceop

* delete duplicate tests

* test_allow_push_permutes

* one kernel tests
 2025-03-09 15:01:08 +01:00
qazalandGitHub 286b480f82 do not replace assign with the offset buffer [pr] (#9387) 2025-03-08 11:57:44 +01:00
qazalandGitHub 0d2762c010 prep refactor for adding buffer ops last [pr] (#9383) 
* prep refactor for adding buffer ops last [pr]

* freeze buffers

* add swizzle_reduceop

* shape for reduceop_view_right

* simpler elementwise_view_right

* add shapetracker to const

* only const

* from process replay
 2025-03-08 08:00:14 +01:00
nimlgenandGitHub 243078dda9 am: optimize tlb usage (#9049) 
* am: optimize tlb usage

* fxies

* comments

* tiny
 2025-03-07 19:37:29 +03:00
geohotstanandGitHub 088d86691b fix onnx gather and onnx auto_pad VALID mode (#9375) 
* fix gather and auto_pad

* long -> int64
 2025-03-07 10:27:23 -05:00
hoovedandGitHub 136cf7b8b1 hotfix: load >2 GiB from disk on macOS (#9361) 
* enable loading >2 GiB buffer from disk on macOS

* handle None case raised by mypy

* add test

* revert fix to repro bug in CI

* tell CI to run a unit test for macOS

* reapply fix
 2025-03-07 14:51:58 +08:00
nimlgenandGitHub 9bd13de44c lower test_gemv_4096_16384 to 750 for red (#9367) 2025-03-05 22:44:48 +03:00
uuuvnandGitHub b75f307234 amd: autogen ip bases (#9360) 2025-03-05 22:30:38 +03:00
chenyuandGitHub 2cb2fce8d9 lower test_gemm_8192 amd_tflops to 65 (#9364) 2025-03-05 14:06:11 -05:00
nimlgenandGitHub 14c88abf27 add some options to allreduce bench (#9348) 2025-03-04 23:46:36 +03:00
Anish UmaleandGitHub bafa40fe12 Tiny backend test_ops fix part1 (#9338) 
* extract name methods from https://github.com/tinygrad/tinygrad/pull/9302

* t.grad.numpy() -> t.grad.cpu().numpy()

* revert TORCH_DEBUG change

* revert dtype change in aten.sum
 2025-03-03 12:36:51 -05:00
George HotzandGitHub 0d4ba7dd87 import tinygrad.frontend.torch (#9337) 
* import tinygrad.frontend.torch

* type ignore
 2025-03-04 00:15:29 +08:00
qazalandGitHub 23084fd850 merge merge_views and remove_movement_ops [pr] (#9333) 
* merge merge_views and remove_movement_ops [pr]

* fix that assert
 2025-03-03 12:38:59 +01:00
George HotzandGitHub ece0a0f305 use empty for test instead of rand (#9332) 2025-03-03 16:19:06 +08:00
George HotzandGitHub 2cc4cb74f0 reorder binops (#9328) 
* reorder binops

* test improvements + fix string tests

* ugh, okay this
 2025-03-03 14:58:18 +08:00
chenyuandGitHub 146eb73790 fix Tensor.view with a tuple arg (#9330) 2025-03-02 23:35:23 -05:00
chenyuandGitHub ba4b8c2c23 Tensor.copysign (#9329) 2025-03-02 21:33:49 -05:00
nimlgenandGitHub 8cae00833c flaky test in ci (#9321) 2025-03-02 16:27:22 +03:00
Ali LadjevardiandGitHub 00028e87bb Failing test for not realizing intermediate expand in multi-GPU (#9320) 2025-03-02 12:54:48 +01:00
geohot ba97fd0b9c hotfix: add test/external/external_benchmark_disk_raw 2025-03-02 02:32:15 +00:00
chenyuandGitHub cc2bbb0bf1 Tensor.isfinite (#9316) 2025-03-01 19:58:56 -05:00
geohotstanGitHubchenyu
 		d9ec05cea6 Test Onnx quantization behavior (#9301) 
* add DynamicDequantizeLinear and corresponding tests

* wow qlinearops are round away from zero

* this passes locally...

* again

* try

* try separate test

* round to even again

* also add QLinearMul

---------

Co-authored-by: chenyu <[email protected]>
 2025-03-01 19:21:58 -05:00
chenyuandGitHub fe0f860209 update test_ops for tensors from torch (#9308) 
a few detach().numpy() -> detach().cpu().numpy()
 2025-02-28 15:57:25 -05:00
chenyuandGitHub 38d7aae3b7 onnx fmod (#9307) 2025-02-28 14:09:22 -05:00
chenyuandGitHub 7c7db78feb support float mod (#9306) 
also added spec check on Ops.MOD to be ints only
 2025-02-28 13:33:58 -05:00
chenyuandGitHub 90808e2dd0 div rounding_mode (#9304) 2025-02-28 11:38:25 -05:00
chenyuandGitHub 3ae66e59a3 least_upper_float is at least default_float (#9303) 
* least_upper_float is at least default_float

en route for div rounding mode. dtype of true int division would change from int32 to default_float, which matches torch too.

* fix bert acc
 2025-02-28 10:41:56 -05:00
Eitan TurokGitHubchenyuGeorge Hotz
 		d657d5f754 [Bounty] Vectorize Transcendental (#9058) 
* init

* cast everythig right

* more casting

* install pillow in test

* quick tests

* simplify

* quick tests

* delete test

* tests

* fix import error

* add vec to ldexp3k

* vec for bitcast

* some helper tests

* high level tests

* clean tests

* change tolerance so cuda passes

* ruff passes

* remove tests for transcendental helpers

* ruff passes

* make exponent in power vectorized

* fix pow test

* add newline

* add vec dtype to ilogb2k

* comment + clean up

* ruff

---------

Co-authored-by: chenyu <[email protected]>
Co-authored-by: George Hotz <[email protected]>
 2025-02-28 15:47:25 +08:00
qazalandGitHub cdf66cc67f test: recompute expanded CAST (#9286) 
* those views should merge

* diff cleanup

* gpu

* put it behind CAST_AFTER_EXPAND
 2025-02-27 19:22:17 +01:00
chenyuandGitHub 4342300eff lower test_gemm_8192 amd to 70 (#9277) 
flaky
 2025-02-26 16:32:08 -05:00