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
11,999 commits 280 branches 15 tags 325 MiB
Commit graph

188 commits

Author SHA1 Message Date
chenyu
26f5c00265
 		move TestTensorMetadata to unit (#14442) 2026-01-30 12:14:21 -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
chenyu
efcb32f6a9
 		unique const when requires_grad is set to True (#14075) 
* unique const when requires_grad is set to True

* fix pyrender
 2026-01-08 16:30:45 -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
chenyu
ed962786d6
 		use assign in Tensor.backward (#13674) 
preserve the grad object so that jit works
 2025-12-13 22:43:06 -05:00
George Hotz
c5bd28e21d
 		start work on schedule cache (#13529) 
* start work on schedule cache

* local unique

* schedule cache works

* schedule cache cleanup

* fix tests

* preserve metadata

* oops, fix cache

* put that there

* fix spec

* always miss

* why is that broken?

* src[0].op

* fix process replay

* delete abstractions2

* reenable the actual schedule cache

* metadata is best effort

* fix JIT in examples/gradaccum_mnist.py

* full jit

* fixed and test is real
 2025-12-04 17:24:49 -08:00
George Hotz
24ca8eeaa7
 		small fixups from schedule_cache (#13557) 2025-12-03 15:41:16 -08:00
Ayman Jabr
256f81bb02
 		Fix tracemeta 0 (#13049) 
* chore: tclesius branch resolved

* fix: indentation

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-11-13 09:07:11 -08:00
George Hotz
b147e7e8e6
 		flatten bufferize (#12984) 
* flatten bufferize

* simpler

* tests pass

* flat

* not flat
 2025-10-29 11:23:43 +08:00
George Hotz
b0da173f2f
 		add unique to const, fix longstanding bug (#12965) 
* add unique to const, fix longstanding bug

* _force_unique=True

* fix tests

* fix more tests
 2025-10-28 15:11:37 +08:00
Sieds Lykles
7f798a9630
 		Cleanup const buffers (#12829) 
* split pm_cleanups

* update test_schedule

* shrink when we remove bufferize

* dont do shrink if shape is empty

* update tests

* remove *1 from metadata

* deal with the noop bufferize

* only noop on cvar

* cleanup

* fix if

* rename
 2025-10-21 14:53:49 +02:00
George Hotz
8be7844b2e
 		use apply uop for assign to fix assign metadata (#12732) 
* use apply uop for assign

* fix metadata for assign

* fix backward metadata

* those aren't real tests
 2025-10-16 20:34:12 +08:00
George Hotz
592e86f6f5
 		remove UOp.st (#12716) 
* remove UOp.st

* fix tests

* torch backend disable
 2025-10-16 14:44:09 +08:00
chenyu
c3278e5622
 		clean up old tests (#12708) 2025-10-15 17:53:17 -04:00
chenyu
312c622d35
 		support None in pad_to and shrink_to (#12700) 2025-10-15 09:25:31 -04:00
Christopher Milan
0aabc1e938
 		Mesa NIR backend (NAK/LLVMpipe) (#12089) 
* nak works

* TestOps::test_add works

* testop has no crashes

* fix bool casts

* fix typo

* add disassemble

* RANGE and locals/regs

* simplify NAKCompiler

* disass cleanup

* cleanup nir codegen

* almost all tests passing

* cleanup notes in extra/

* old notes

* only import nak if NIR=1

* fix new SPECIAL syntax

* fix local/shared memory

* more tests passing

* add DEFINE_VAR support

* llvmpipe kinda works

* diskcache

* some mypy stuff

* lvp passing test_ops.py

* fix imports

* actually fix imports

* remove 'stdout'

* fix llvm import

* fix mypy issues

* nicer errors

* simpler test_dtype skips

* test lvp in CI

* fix github action syntax

* fix more actions typos

* switch to mesa 25.1.0

* diskcache_put

* better generation for lvp nir_options

* b64encode shader blobs

* Revert diskcache changes

This reverts commits 930fa3de8a and 8428c694b3.

* general cleanup

* better error messages

* fix llvm import

* fix windows tests

* link with libm and libgcc_s

* fix some errors

* dont check for 'float4'

* NIR uses pointer arithmetic

* use tinymesa

* bump tinymesa

* bump tinymesa again

* update lvp nir_options

* print nir shader with DEBUG

* simplify LVPCompiler

* more tests

* "gated" STORE

* NAK is cacheable

* more tests

* all tests pass locally for NAK

* test autogen in CI

* autogen deps

* more deps

* fix uop_gc

* fix macos

* mypy

* save 2 lines

* save two more lines

* save 1 line

* save 4 lines

* save more lines

* Revert "save more lines"

This reverts commit dd3a720c5a.

* save more lines

* fix LVP on windows

* refactor

* reorganize some code

* refactor lib_gpu

* move LVP check

* out of order loads

* remove support.mesa

* bump tinymesa version

* simplify LVP jit

* macos

* macos ci

* shell: bash

* testing

* more testing

* compute brew prefix

* stupid typo

* actually fix

* lib

* stdout on macos

* inline gallivm_compile_module

* Revert "inline gallivm_compile_module"

This reverts commit b65983b151.

* elf macos

* semicolon

* inherit from CPULLVMCompiler

* ruff

* disas test

* fix libm linking

* default is fine actually

* arm works

* add elf loader link test

* fix NAK beam

* pylint is too smart by half

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
Co-authored-by: nimlgen <138685161+nimlgen@users.noreply.github.com>
 2025-10-15 17:38:33 +08:00
George Hotz
db4a359374
 		fix up some slow tests that launch python (#12672) 
* fix up some slow tests that launch python

* svd nonfull in parallel

* split test_advancedindex
 2025-10-14 19:13:55 +08:00
chenyu
c8dfd10257
 		ShapeTracker.real_strides -> is_expanded [pr] (#12579) 
only keep the used part
 2025-10-09 22:52:45 -04:00
chenyu
ae51bdd06a
 		remove trivial use of RANGEIFY flag (#12550) 
some tests need update still
 2025-10-09 02:29:38 -04:00
George Hotz
3b0b3a2e64
 		fast RANGEIFY (#12504) 
* rtoposort is fast, can replace rangeify with this

* fast rangeify

* work

* fast rangeify works for mnist

* should work

* progress

* pad fix

* FAST

* tests passing

* don't delete those shape ops

* put in rangeify map

* ending ranges fix

* tests

* mstack/mselect no hacks

* move to indexing.py

* touch up tests + add comments

* disable failing test

* actually make the file readable

* failing

* error
 2025-10-08 19:38:06 +08:00
chenyu
f82b16a0e9
 		RANGEIFY test_tensor (#12235) 2025-09-18 10:35:43 -04:00
chenyu
84d2d047ea
 		Tensor.pad_to and Tensor.shrink_to (#12210) 
most of the time i want this instead of spelling out the args

also add more input validation to shrink
 2025-09-16 12:24:55 -04:00
Sieds Lykles
2fc0bd150b
 		Arange overflow raises error and one_hot upcast (#11975) 
* add error

* to_dtype

* shorten line

* add test

* upcast one hot dim im overflows
 2025-09-13 00:18:25 +02:00
chenyu
647965fb09
 		test_train cleanup (#12140) 
* test_train cleanup

remove skipIf due to buffer sizes, runs locally

* those are slow
 2025-09-12 13:21:30 -04: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
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
chenyu
337e979a59
 		call dtypes.as_const in Tensor(list) (#11840) 2025-08-25 22:08:26 -04:00
George Hotz
07b0df0d86 hotfix: test tensor dims start at 1 2025-08-05 15:40:24 -07:00
George Hotz
4dabdf7c6d
 		Revert "optimize in rewrite (#11516)" (#11517) 
This reverts commit 3b777a9e05.
 2025-08-05 15:39:07 -07:00
George Hotz
3b777a9e05
 		optimize in rewrite (#11516) 
* changes

* fix test uops

* dim shouldn't be 0

* huh, why did that one not save
 2025-08-05 15:33:26 -07:00
chenyu
5b570196e4
 		support DEV= to specify device (#11351) 2025-07-23 17:40:55 -04:00
George Hotz
32e9949052
 		rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07:00
George Hotz
4c315f8e17
 		MSTACK little non-functional changes (#10648) 2025-06-05 13:20:22 -07:00
George Hotz
b3b43a82c4
 		remove Tensor.no_grad, it's meaningless now [pr] (#10556) 2025-05-28 22:20:02 -07:00
George Hotz
b58f2d4544
 		fix tests (#10493) 2025-05-23 18:38:07 -07:00
George Hotz
1e4d63e06e
 		uops can have multiple metadata (#10479) 
* uops can have multiple metadata

* fixups
 2025-05-22 21:35:02 -07:00
chenyu
8cc2dff4d8
 		only float Tensors have gradient [pr] (#10475) 2025-05-22 21:02:11 -04:00
George Hotz
411392dfb7
 		move files into uop dir (#10399) 
* move files into uop dir [pr]

* tinygrad.uop is a thing

* fix uop docs, no pr

* fix viz
 2025-05-18 11:38:28 -07:00
George Hotz
603c03bef2
 		fix tests for rewrite [pr] (#10167) 
* fix tests for rewrite [pr]

* cleaner

* delete linearize_uop

* clean up the rest
 2025-05-05 19:19:49 -07:00
Park Jun
c3ad7b2a84
 		create randperm and support pytorch backend (#10019) 2025-04-24 07:29:02 -04:00
George Hotz
e358e0a0c6
 		move metadata set to tensor [pr] (#9976) 
* move metadata set to tensor [pr]

* only track that in tensor.py
 2025-04-22 12:30:35 +01:00
Andrew Furey
50dee4a7b3
 		add test for checking const gradients (#9598) 2025-03-27 15:17:37 -04:00
chenyu
22fc0a2e36
 		bert sum acc in half (#9412) 
also BS=96
 2025-03-11 23:03:15 -04:00
qazal
81a71ae0f6
 		hotfix: skip test_exclude_const_metadata (#9208) 2025-02-22 23:26:04 +02:00
qazal
4578c3e8fd
 		simpler tensor metadata mapping + tests [pr] (#9203) 
* simpler tensor metadata mapping + tests [pr]

* remove kernel metadata

* don't map nones
 2025-02-22 20:18:46 +01:00
chenyu
2e7c2780a9
 		CLANG -> CPU (#9189) 2025-02-20 18:03:09 -05:00
George Hotz
a4dab3ec3f
 		add name uop (#9149) 
* add name uop, TODO: refactor renderer to use

* renderer uses name uop

* fix tests

* render

* ptx
 2025-02-18 15:26:58 +08:00
George Hotz
df3b320f46
 		rewriter -> devectorizer [pr] (#9147) 2025-02-18 12:42:08 +08:00
Josh Moore
44e0eab8fd
 		Fix AttributeError occurring after ValueError in _apply_uop (#8905) 
* Fix AttributeError occurring after ValueError in _apply_uop

* Update tensor.py

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-02-06 10:56:29 +08:00
chenyu
1f730ae8f8
 		remove retain_graph in Tensor.backward [pr] (#8835) 
not used. gradient accumulation works directly
 2025-01-31 13:41:26 -05:00