mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
13,471 commits 280 branches 15 tags 325 MiB
Commit graph

5,694 commits

Author SHA1 Message Date
chenyu
520e2e0727
 		actually run unit tests in ci MacOS (unit) (#12122) 
* actually run unit tests in ci MacOS (unit)

* that's always wrong
 2025-09-11 13:32:30 -04:00
chenyu
20cd7177de
 		delete test_bert_fuse_arange (#12121) 
* delete test_bert_fuse_arange

it's the default now and we are not interested in FUSE_ARANGE=0 version

* remove -v
 2025-09-11 12:35:51 -04:00
George Hotz
3ef0e5e01e
 		rangeify: use Ops.REALIZE and not Ops.CONTIGUOUS if it's added by system (#12111) 
* rangeify: use Ops.REALIZE and not Ops.CONTIGUOUS if it's added by system

* fix contig + BufferizeOpts

* no outerworld
 2025-09-11 11:56:59 +08:00
b1tg
52ebed991e
 		change dtype promo lattice when fp8s is supported (#12088) 
* change dtype promo lattice when fp8s is supported

* no device check

* int64 + uint64 => fp8
 2025-09-10 22:09:11 -04:00
George Hotz
d4eba5800d
 		rangeify cost function infrastructure (#12091) 
* one call to hc opt

* does that pass?

* add cost function to rangeify

* test

* more test

* gate thread

* bufferize has shape

* ish

* match old behavior

* no ci there
 2025-09-11 07:19:53 +08:00
Sieds Lykles
3989f5b559
 		Revert "Simplify valid in symbolic (#12104)" (#12108) 
This reverts commit 73d479a016.
 2025-09-10 23:36:40 +02:00
Sieds Lykles
73d479a016
 		Simplify valid in symbolic (#12104) 
* cleanup cast_folding

* from sym to symbolic

* no more sym in dtype lowering

* move around simplify_valid

* update test
 2025-09-10 23:26:19 +02:00
chenyu
0e266f376c
 		ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
nimlgen
fb96394ff5
 		auto-select available compilers (#12094) 
* device: auto select compilers

* fix

* metal+opencl

* nv/cuda

* test without ptx

* ptx

* fix tests

* fix

* fix test

* rename

* test + cleaner

* xx

* ops

* better test

* win?

* um?

* types

* debug

* win??

* sep rung

* wtf?

* debug

* skip win

* revert this

* types
 2025-09-10 19:52:01 +03:00
George Hotz
9789337722
 		early reduce simplify (#12046) 
* early reduce simplify

* min changes

* need that

* that goes in simplify

* no more arange reduce opt
 2025-09-10 21:02:46 +08:00
nimlgen
551560b87c
 		do not use getenv('PTX') in tests (#12095) 
* test without ptx

* fix tests

* fix test

* linters
 2025-09-10 14:04:07 +03:00
Sieds Lykles
0e420e68b4
 		delete axis_is_masked (#12092) 2025-09-10 05:26:19 +02:00
Sieds Lykles
499f50483b
 		x | !x -> True (#12090) 2025-09-10 03:26:01 +02:00
Sieds Lykles
5b73076e48
 		assert benchmark times (#12042) 
* assert jitted times in openpilot

* better error

* better error

* add ASSERT_MIN_STEP_TIME to more models

* t is step_times

* update benchmark times

* update times
 2025-09-09 23:40:02 +02:00
b1tg
58d13a6e3e
 		remove redundant check (#12087) 2025-09-09 15:15:39 -04:00
b1tg
82e955fe79
 		fix inf bug in float_to_fp8 (#12085) 2025-09-09 12:02:56 -04:00
b1tg
14faf7a5c0
 		AutoCastType tests for fp8s/bf16 (#12084) 2025-09-09 11:33:01 -04:00
nimlgen
1c6c42715f
 		unify cpu and llvm (#11982) 
* try unify cpu and llvm

* fixes

* fix

* ops

* no llvm

* fix

* rm

* lvmm is ot

* oops

* override

* no llvm

* ignore

* skip llvm

* ooops
 2025-09-09 13:54:44 +03:00
nimlgen
9182948951
 		remove llvm_bf16_cast (#12075) 2025-09-08 20:51:15 +03:00
Sieds Lykles
75b58fe2d3
 		move simplify_valid pat to sym (#12065) 
* move simplify_valid pat to sym

* fix expectedfailure
 2025-09-08 07:01:26 +02:00
chenyu
56861852be
 		enable IMAGE for test_mnist and test_mnist_backward (#12064) 
passes now
 2025-09-07 09:06:39 -04:00
nimlgen
10ac427aaa
 		cpu threading (#11951) 
* start cpu threading

* fix

* fix2

* fix

* hacks?

* threads

* minor

* no dsp

* dsp 2

* n

* more

* test

* xm

* cleaner

* readable

* f

* reorder

* when no threads

* rangeify

* typos

* not needed

* reapply

* remoev this

* linter

* fixed cpu count in ci

* fix

* fixes

* rm

* typo

* sort based on speed

* test if test works in ci

* Revert "test if test works in ci"

This reverts commit 1f05edb531.

* do not pad thread
 2025-09-06 16:13:43 +03:00
Sieds Lykles
581b2388c2
 		add dtypes.index (#12015) 
* add dtypes.index

* cast shape, stride and mask to dtypes.index in view.create

* move pm_lower_index_dtype to ops

* DEFINE_VAR is dtype.index by default

* merge var_val_using_str

* remove int from commutative

* fix test_rewrite_map

* change that to dtypes.index

* change some int to index

* shorten those

* remove old cast in renderer

* cleanup

* change that back

* add comment

* delete comment

* just delete those

* view doesnt have to cast anymore

* adjust comment
 2025-09-06 06:03:44 +02:00
Sieds Lykles
c6c16b2946
 		var_vals uses str for var (#12011) 
* var_vals is str,int

* remove imports

* remove print

* fix test

* change var_vals in hcq

* update test_hcq

* fix multitensor _device_num var

* fix syminfer test

* shorten line

* p.vars stays list[Variable]

* shorten line

* vars is back to tuple[Variable, ...]

* change var_vals in extra

* change var_vals from shapetracker

* var_vals is str:int

* fix signature
 2025-09-06 04:16:12 +02:00
George Hotz
38dcadf07b
 		delete kernel.py (#12040) 
* delete kernel.py

* delete that file

* rip and tear

* don't test search

* imports

* fix torch frontend

* not a part of regen
 2025-09-05 15:52:07 -07:00
George Hotz
ee4f696086
 		delete more tests (#12043) 
* delete more tests

* delete and simplify

* flaky on windows

* a few more, those remained
 2025-09-05 15:31:30 -07:00
George Hotz
12c7b1bb01
 		cleanup lin tests without Kernel (#12041) 
* cleanup lin tests without Kernel

* no kernel.py there

* remove that test
 2025-09-05 15:13:14 -07:00
George Hotz
433581f8ed
 		make POSTOPT=2 the default (#12034) 
* make POSTOPT=2 the default

* more matching tc

* fix winograd

* fix that test

* add matvec to Scheduler

* flip tc sort order

* similar speed

* fix beam on image

* disable slow tests

* slow
 2025-09-05 14:34:05 -07:00
chenyu
3b41a04b96
 		remove test_openpilot in test_onnx (#12037) 
openpilot is tested in compile3
 2025-09-05 16:20:03 -04:00
George Hotz
870f63d9cc
 		add WARP axistype, fix postopt bugs (#12033) 
* postopt is 83% match

* warp is bright CYAN

* beautiful mnist beam works

* fix shutdown bug
 2025-09-05 10:36:55 -07:00
chenyu
4c2d4f683a
 		lower universal_test_unary cos domain (#12032) 
flaky
 2025-09-05 12:19:44 -04:00
chenyu
ce7163e9b4
 		clean up skip slow tests in PYTHON (#12028) 
skip with SKIP_SLOW_TEST and decorators
 2025-09-05 11:35:26 -04:00
chenyu
e0da644171
 		lower sample count in test_multinomial (#12027) 2025-09-05 10:10:28 -04:00
chenyu
9b6f1b86cb
 		add Tensor.maximum in test_dtype_alu (#12025) 
works except nan
 2025-09-05 09:48:39 -04:00
qazal
f461ccf407
 		exclude op2 nan lt in test_dtype_alu (#12024) 
failure: https://github.com/tinygrad/tinygrad/actions/runs/17490320000/job/49679581331?pr=12022#step:6:125
 2025-09-05 14:14:22 +03:00
George Hotz
2b5a73ac65
 		improve test_linearizer (#12016) 
* improve test_linearizer

* tweaks

* simpler

* get_prg

* that one doesn't have to return

* fix postopt bugs

* fix rng
 2025-09-04 20:44:05 -07:00
chenyu
7f3df6ea21
 		exclude nan in test_dtype_alu lt (#12019) 2025-09-04 23:38:37 -04:00
Sieds Lykles
f5404ca53c
 		Divmod combine - associative variations (#12017) 
* add rule and test

* more rules and tests

* add all four variations

* fix test

* test fixed!

* adjust commment

* add new variations

* disable intel tensor core ops count test for bigger_matmul_half
 2025-09-05 03:44:02 +02:00
chenyu
677220ae7e
 		test_tesnor_data to unit/ (#12013) 2025-09-04 19:58:27 -04:00
George Hotz
30eb42a69e
 		fix POSTOPT pad (#11999) 
* fix POSTOPT=1

* fix some tests

* Revert "fix some tests"

This reverts commit 8ee058e206.

* fix padding restrictions

* cuda has two tensor cores

* Set POSTOPT ContextVar to 0 in helpers.py
 2025-09-04 14:28:58 -07:00
qazal
da61b40604
 		some viz tests don't need track_rewrites (#12010) 2025-09-04 23:59:32 +03:00
qazal
be364a1adb
 		viz: add default tracing group (#12009) 
This enables seeing rewrites in unit tests like `VIZ=1 python3 test/test_uop_graph.py TestUOpGraph.test_in_bounds_access_gated_local` that call graph_rewrite directly.

`@track_rewrites` keeps existing as an optional helper to organize larger traces.
 2025-09-04 23:29:56 +03:00
chenyu
52166fd7eb
 		smaller test_ops inputs (#12007) 2025-09-04 16:22:33 -04:00
chenyu
dc8501af30
 		clean up wino tests (#12008) 
removed the one that tests hcopt and added one for backward kernel counts
 2025-09-04 16:14:55 -04:00
chenyu
8c720e8760
 		less iterations for symbolic double for loops (#12006) 2025-09-04 15:09:17 -04:00
George Hotz
70ce29b630
 		test pyrender (#12005) 
* test pyrender

* make them print

* switch to pyrendered
 2025-09-04 11:48:40 -07:00
George Hotz
560df206cc
 		split tc test (#12003) 
* split tc test

* split hand coded opts

* remove some skipped tests

* skips on emulated
 2025-09-04 11:47:56 -07:00
qazal
4996bb668b
 		load all traces before asserting in test_viz (#12004) 2025-09-04 21:34:48 +03:00
George Hotz
9dee724fc4
 		make EMULATE a context var (#12002) 
* make EMULATE a context var

* fix test amx
 2025-09-04 11:15:43 -07:00
George Hotz
09106e4aae
 		refactor and split test_linearizer (#12001) 
* refactor and split test_linearizer

* forget that file

* imports

* remove from docs

* test gen float4
 2025-09-04 10:53:07 -07:00