sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 4 Activity
14,300 Commits 336 Branches 15 Tags
simple_rangeify
Commit Graph
46 Commits
Author SHA1 Message Date
Robert JosephandGitHub 0c96cdc300 fix prod gradients at zero (#17404) 2026-08-07 09:56:52 -04:00
chenyuandGitHub d79772f057 fix pow on extreme inputs (#17397) 
* fix pow on extreme inputs

* WEBGPU
 2026-08-04 19:30:43 -04:00
Noah SchiroandGitHub f2c2f4456b Add softmin (#17292) 
* Add softmin

* Remove extra tests
 2026-07-30 20:05:11 -04:00
chenyuandGitHub acc2374b6f minor normalize cleanup [pr] (#17215) 
and add a few no-op clones in test_ops to prep for weak flip
 2026-07-26 00:39:53 -04:00
chenyuandGitHub a292dab8ab corner cases from weak const branch (#17173) 2026-07-24 00:09:56 -04:00
chenyuandGitHub f5d9c31d15 arange upcast to int64 with big N (#17167) 
simplifies _one_hot_along_dim
 2026-07-23 20:11:39 -04:00
chenyuandGitHub 3e031d6c4d AND OR XOR SHL SHR cannot have float operands [PR] (#17164) 
* AND OR XOR SHL SHR cannot have float operands [PR]

* dont
 2026-07-23 18:51:56 -04:00
chenyuandGitHub 2983987321 clean up some tests (#17154) 
duplicated or typo or slow
 2026-07-23 11:22:18 -04:00
sirhcmandGitHub c7027db715 lvp: skip test_masked_select (#17137) 2026-07-22 14:53:09 -04:00
chenyuandGitHub 92f9c850b4 fix pow(int, float) (#17126) 
* fix pow(int, float)

* onnx
 2026-07-21 18:48:24 -04:00
chenyuandGitHub f19a2ad771 single where mixin [pr] (#17118) 
* single where mixin [pr]

no shape broadcasting in ufix and _broadcasted anymore

* QCOM vectorized bool is broken
 2026-07-21 17:38:36 -04:00
chenyuandGitHub 9970a0aad0 fix Tensor << Tensor for x86 (#17082) 
* fix Tensor << Tensor for x86

* torch
 2026-07-19 12:31:05 -04:00
sirhcmandGitHub 353d8f1e13 use rusticl in ci (#16852) 2026-07-03 23:58:58 -04:00
chenyuandGitHub 59c7874724 fix isclose with scalar other (#16813) 2026-06-30 21:56:33 -04:00
Javier De JesusandGitHub 989f713c1b support negative pads in circular pad mode (#16448) 2026-05-31 09:28:45 -07:00
sirhcmandGitHub 8ddd1328df remove getenv(CI) (#16365) 
gone everywhere except test_interop, because torch MPS does not work in actions
 2026-05-25 20:23:33 -04:00
sirhcmandGitHub d8f86be613 webgpu: shader-f16 support in arch (#16370) 2026-05-25 19:20:59 -04:00
chenyuandGitHub 926d125a63 update test_stack (#16345) 
also skip COMPILE_ONLY, it was comparing 0==0
 2026-05-23 10:42:35 -04:00
sirhcmandGitHub 451f38155c start cleanup of the slowest tests (#16339) 2026-05-22 18:39:36 -04:00
sirhcmandGitHub c2d06570a5 remove getenv(CI) from core tinygrad (#16326) 2026-05-21 22:20:33 -04:00
sirhcmandGitHub 172f9493e1 move is_dtype_supported to renderer (#16226) 2026-05-20 21:19:37 -04:00
George HotzandGitHub 58d58c1659 remove DEVECTORIZE (#16290) 
* remove DEVECTORIZE

* fully remove DEVECTORIZE
 2026-05-20 13:25:49 -07:00
George HotzandGitHub a120709671 tighten shape spec for broadcasting (#16206) 
* tighten shape spec for broadcasting

* use IndexError, not ValueError

* needs size
 2026-05-18 22:12:04 -07:00
chenyuandGitHub 8631b6f17d remove use of requires_grad in test/ (#16237) 2026-05-16 17:21:07 -04:00
C TandGitHub 1b779a9058 add gelu approximate="none" (match pytorch) (#16162) 
* add gelu approximate="none" (match pytorch)

* lint

* pass through onnx Gelu approximate

* type annotate

* explicit math.sqrt

* keep tinygrad's gelu approximate="tanh" default
 2026-05-13 18:53:24 -07:00
chenyuandGitHub bdcdf1f1a1 jittable masked_select and nonzero (#16170) 
* jittable masked_select and nonzero

make jittable with `size=`, matches jax

* COMPILE_ONLY
 2026-05-12 16:39:36 -04:00
chenyuandGitHub 7c3e3fa154 fix empty input for masked_select and nonzero (#16168) 2026-05-12 15:36:51 -04:00
PawanandGitHub 4dd6ad3514 gradient: add TRUNC backward (#15925) 
* gradient: add TRUNC backward

* test: move round quantization gradient to test_ops
 2026-05-08 16:27:55 -07:00
chenyuandGitHub 072db9924c div to mixin (#16078) 
also deleted idiv method
 2026-05-07 12:52:37 -04:00
chenyuandGitHub 782d1ff80f Tensor.fmod (#16014) 
c-style mod matches torch
 2026-05-01 16:02:18 -04:00
chenyuandGitHub f911a63a6b don't allow negative num_classes in one_hot (#15859) 
no auto infer num_classes, matches jax
 2026-04-21 19:39:29 -04:00
sirhcmandGitHub 6adf4c3cd9 MOCKGPU interfaces (#15796) 2026-04-17 21:56:29 -04:00
chenyuandGitHub 0191cc73dc update arange range check (#15794) 
it was not checking negative steps correctly
 2026-04-17 16:07:50 -04:00
chenyuandGitHub 01b49c8647 support int operand for shifts (#15618) 
matches torch/jax, also symbolic rule to remove mask
 2026-04-06 12:32:12 -04:00
sirhcmGitHubComma Device
 		645d45d968 DEV has arch (#15577) 
Co-authored-by: Comma Device <[email protected]>
 2026-04-03 19:17:19 -04:00
sirhcmandGitHub acf239e4d2 specify renderer in DEV, <dev>_<ren>=1 is deprecated (#15551) 2026-03-31 18:35:14 -04:00
sirhcmandGitHub bc180a963c deprecate <dev>=1 in favor of DEV=<dev> (#15467) 
* start work on target

* add test

* update actions to use DEV

* update docs

* update readmes

* tests need that too

* update example

* update tests (comments)

* fix that test

* ruff

* mypy

* oops

* remove getenvs

* don't add Target yet

* and the test

* lint

* and docs

* more stuff

* assert

* few more fixes

* test assert
 2026-03-26 03:48:03 -04:00
chenyuandGitHub 842c978df3 remove staticmethod dtypes.max/min (#15227) 
always use x.dtype.max/min
 2026-03-11 23:11:24 -04:00
sirhcmGitHubComma Device
 		7810be8d3c compile QCOM without opening device (#15165) 
Co-authored-by: Comma Device <[email protected]>
 2026-03-06 06:24:27 -05:00
Roelof van DijkandGitHub d65923bda5 tensor.py: add normalize function (#15159) 
* tensor.py: add normalize function

* p==0 should match torch
 2026-03-05 18:55:53 +08:00
chenyuandGitHub 127136421d enable a few WEBGPU isnan tests that work now (#14967) 
* enable a few WEBGPU isnan tests that work now

* still failed
 2026-02-23 11:06:08 -05:00
George HotzandGitHub 8ef5544e4a realized PYTHON copies (#14934) 
* realized PYTHON copies

* comment that out

* fix that test

* append afters

* contig

* disk copies

* should be 124

* 332
 2026-02-21 20:29:31 +08:00
George HotzandGitHub 55d3a5def9 preallocate all realized buffers (#14823) 
* preallocate all realized buffers

* contiguous

* work

* comment that out

* move to schedule

* better

* correct fix

* just buffer

* disk bufs

* fixes disk tensor stuff

* fix symbolic stuff

* fix multi

* 162 failures

* bugfixes

* don't check that anymore

* fix schedule tests

* mnist should be contiguious

* type and buffer

* fix tests

* shrink axis correction

* mypy fixes

* tests skips

* same 37 failures

* dedup

* no shrink in the graph

* 29 failures

* skips

* fix custom kernel

* fix training

* those optimizations aren't supported currently

* simpler

* more correct

* tests

* 14 failures

* works

* fix that test

* broken

* 11 failures

* only kernel counts left

* fixes

* all tests pass

* remove tensor_map

* op test

* 200 -> 230

* test fixes

* fixes

* revert test_tiny thing

* guard

* revert that

* test tiny passes

* no contigs there

* base realize back

* Revert "no contigs there"

This reverts commit c45bb9fcfd.

* revert that

* chop many assigns

* 12 failures

* fix tests

* tests

* apply after

* pre-commit

* remove old code

* delete that

* fix types

* remove extra contig

* fix dataloader

* torch fix

* disk fix

* update kernel fusion numbres

* runs on amd

* restore kernel count

* add that rule back

* that

* disable that

* wrong

* add the correct rule for that folding

* more tests

* guard c1.arg

* no newlines

* realize those

* split into a different file

* remove detach/contig back

* skip 2

* update that
 2026-02-20 20:05:54 +08:00
chenyuandGitHub 352845d8cc update cast to uint tests (#14768) 
result in valid range should work, add intermediate cast to NIRRenderer since it's UB for [128, 256)
 2026-02-15 10:55:13 -05:00
George HotzandGitHub 0e215c433d remove hack from cast (#14760) 
* remove hack from cast

* skip tests

* linters to 3.12, another skip

* fix rand

* m_
 2026-02-15 13:56:38 +08:00
George HotzandGitHub c331798201 move tests to test/backend (#14691) 
* move tests to test/backend

* fix imports

* fix CI

* revert that one

* Fix formatting in README for test command
 2026-02-12 11:09:44 +08:00