mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-14 00:45:16 +08:00
Code Issues Packages Projects Releases Wiki Activity
13,471 Commits 277 Branches 15 Tags
python_prg
Commit Graph

60 Commits

Author SHA1 Message Date
George Hotz
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
George Hotz
dc77b3318b move files that pass with NULL=1 to test/null (#14508) 
* move files that pass with NULL=1 to test/null

* fix windows

* cpu 0

* bugfix + durations
 2026-02-03 13:52:36 +08:00
Martin Szewieczek
7010c176cf pre commit: fix path to test_assign.py (#14231) 2026-01-19 13:36:30 -05:00
chenyu
e2987001ee unify pre-commit mypy and ci mypy (#13940) 2025-12-31 17:51:51 -05:00
George Hotz
22f3e7f995 better precommit coverage and faster (#13740) 
* improve pre-commit hook speed and coverage

* remove a few

* lose that
 2025-12-17 13:25:55 -04:00
George Hotz
4027eef264 fix test warnings (#13114) 
* fix test warnings

* precommit passes

* ignore std_mean warning
 2025-11-05 15:06:29 -08:00
George Hotz
c0c24d3a70 cleanup wmma (#12927) 
* cleanup wmma

* fix test_ops failures on android
 2025-10-26 18:26:47 +08:00
George Hotz
84d4589ed4 remove pylint from pre-commit and CI (#12658) 
* remove pylint from pre-commit and CI

* multidevice test is fast

* faster pre-commit

* 8 is faster than 4

* better name

* how did that typecheck?
 2025-10-14 15:39:59 +08: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
a38947b4bb move symbolic and transcendental to uop [pr] (#10771) 2025-06-10 20:51:22 -07:00
George Hotz
81b9c04574 move high level stuff to unit tests [pr] (#10708) 
* move high level stuff to unit tests [pr]

* process replay on unit tests

* fix pr, less compute

* set omp num threads

* set 200MB buffer size limit

* delete junk

* fix tests

* faster

* move test_indexing to unit

* faster
 2025-06-08 14:05:56 -07:00
George Hotz
53ed64e133 ci speed work 1 (#10676) 
* skip a few slow tests

* use a venv for python packages

* create venv

* no user, it's in venv

* ignore venv

* venv

* new cache key

* try that

* this

* version the python cache
 2025-06-07 16:33:11 -07:00
George Hotz
9a3d498d9c with commutative hack, uops can change. fix that (#7266) 
* with commutative hack, uops can change. fix that

* simpler
 2024-10-24 18:50:23 +08:00
George Hotz
de7b9d7c42 improve pre-commit [pr] (#7256) 
* improve pre-commit [pr]

* mypy passes on windows
 2024-10-24 15:38:47 +08:00
George Hotz
4013c9848c don't use tons of memory for tests non CI [pr] (#7209) 
* don't use tons of memory for tests

* fix import and clean up pre-commit

* use pathlib

* no shm on windows

* Revert "use pathlib"

This reverts commit 7c38489820.

* run pre-commit hooks in test

* ugh, fix later
 2024-10-22 15:04:51 +08:00
George Hotz
e441794c4b remove custom op support, we waste time maintaining this (#6991) 
* remove custom op support, we waste time maintaining this

* customop is over
 2024-10-11 14:31:09 +08:00
kormann
7c3b877216 rename uop [run_process_replay] (#5031) 
* rename

* fix unittests

* rename vin

* fix test

* fix type [run_process_replay]

* rm pre commit hook change
 2024-06-18 21:34:05 +03:00
George Hotz
afa9753d39 ruff cleanup (#4594) 
* check editor config

* no editorconfig, it doesn't work

* ruff cleanups
 2024-05-14 21:16:14 -07:00
George Hotz
9425973bc7 docs cleanup and move (#4593) 
* cleanup and move

* docs-legacy is gone

* don't update setup.py
 2024-05-14 20:44:59 -07:00
George Hotz
fd02ab1e8b move disassemblers and openpilot (#4592) 
* move disassemblers and openpilot

* delete junk

* put that in pre-commit

* fixup readme
 2024-05-14 19:30:02 -07:00
George Hotz
8f749ae0eb New docs are in mkdocs (#4178) 
* start mkdocs

* simple docs for tensor

* more docs

* move those back

* more docs

* copy markdown extensions

* docs legacy

* docs building workflow

* fix showcase links

* only that?

* install tinygrad

* add docs to setup.py

* Delete examples/llm.c/data
 2024-04-16 10:59:51 +04:00
chenyu
9e0ebf8979 remove dtype from FlopCounter (#4075) 
the annoying thing to remove all FlopCounter is that for device that does not support local, matmul index alu is huge.
we can remove the dtype first.

sneak in updating `ruff` command to `ruff check`
 2024-04-04 21:23:28 -04:00
George Hotz
629cbc5587 only abstractions 2 (#3947) 2024-03-26 20:02:18 -07:00
chenyu
6c1063ba39 add mypy --strict-equality to pre-commit (#3458) 
matched ci mypy behavior
 2024-02-21 03:41:05 -05:00
chenyu
6efa68f97b remove use of TORCH in pre-commit (#3424) 
it's silently using DEFAULT after removing TORCH
 2024-02-15 19:38:37 -05:00
George Hotz
1f9aee8b6f remove numpy from device (#3123) 
* remove numpy from device

* fix tests

* np item

* cleanups

* simplify with as_buffer

* no toCPU

* tinygradic

* cast to scalar
 2024-01-14 19:36:05 -08:00
George Hotz
ea5824657d move fromcpu out of lazy.py (#3122) 
* move fromcpu out of lazy.py

* fix abstractions2
 2024-01-14 18:21:08 -08:00
George Hotz
00d9eda961 FROM -> COPY, move vars_from_ast (#2675) 2023-12-07 16:32:30 -08:00
George Hotz
232ed2af3f more test cleanups (#2631) 
* more test cleanups

* move test example back
 2023-12-05 16:17:57 -08:00
George Hotz
ec594cf03c hotfix: tasteful ctrl-c in parallel beam 2023-12-05 18:20:10 +00:00
George Hotz
5068e99d18 refactor to remove extra kernel params (#2563) 
* refactor to have compiled kernel

* bugfixes

* docs/beautiful.py

* revert that

* fix tests
 2023-12-02 00:32:25 -08:00
George Hotz
f5de21e753 fast path for copy (#2548) 
* fast copy

* ruff first

* flat_mv on malloc

* order + webgpu test
 2023-12-01 11:34:47 -08:00
George Hotz
8fd8399437 remove flake8 (#2544) 2023-12-01 09:48:41 -08:00
chenyu
7d26452305 call ruff with --preview (#2522) 
some checks are ignored without --preview
 2023-11-30 13:59:00 -05:00
George Hotz
c42d2c4731 strip whitespace 2023-11-25 14:09:06 -08:00
George Hotz
70a65c201e JIT support in Interpreted (#2314) 
* factor that out

* jit is supported everywhere

* fix some tests

* there's no jit supported device, the jit is everywhere

* fix test uops
 2023-11-15 11:13:38 -08:00
George Hotz
0cbf6c1811 move things, clean up extra (#2292) 
* move things

* idk why pylint needs that now

* delete unused
 2023-11-13 20:18:40 -08:00
George Hotz
85d26ddc36 uops loop removal (#2262) 
* remove the loop

* cleanups

* tests failing still

* global_loop_ctx wasn't needed

* replace_op is cleaner

* minor opt

* cast opt was wrong

* uop_num

* uop num was dumb

* tuplize_uops

* torch tests

* fix test_uops
 2023-11-10 15:24:47 -08:00
George Hotz
c0f447d6f7 Inline barrier (#2255) 
* put barrier inline for locals

* fix pre-commit on m3

* gate if through barrier
 2023-11-10 08:17:10 -08:00
George Hotz
80bf0b8586 proper wmma (#2245) 
* proper wmma

* hip cast

* bugfixes

* bugfix

* that bug is fixed

---------

Co-authored-by: George Hotz <george@tinygrad.org>
 2023-11-09 15:15:18 -08:00
George Hotz
8932816816 remove arm64, caching for cuda (#2201) 
* remove arm64, caching for cuda

* caching in llvm

* switch cache_compiled to new cache

* fix clang

* caching for metal

* fix pylint

* cleanups

* perf_counter and binary
 2023-11-01 18:44:00 -07:00
qazal
e2428b63a6 external (#2191) 2023-10-31 13:57:24 -07:00
George Hotz
f54959e5cd move print tree into graph (#2003) 
* move print tree into graph

* add winograd profiling test

* change pre-commit to run ruff first
 2023-10-07 04:39:21 -07:00
Vidhan Bhatt
94b21c41a7 ci: use mypy.ini (#1993) 2023-10-06 01:45:28 -07:00
George Hotz
adab724caa schedule2, keep the tests working with small changes (#1932) 
* lazy cleanups

* ast functions take in LazyOps

* op instead of self.op

* _base for mops

* fix contiguous

* start schedule

* test_schedule

* fix openpilot

* more tests

* bugfix and test skip

* work

* make sure things get freed

* fix zerosized tensors

* fix failing test

* fix ceil and friends

* fix openpilot

* disable training

* disable test collectives
 2023-09-28 09:14:43 -07:00
waifairer
d89fb729e5 flake8 (#1323) 
* flake8: Ignore frequent violations, correct infrequent ones

* Ignore some rules in test

* Reorder test ignores

* Lint test + main

* EOF indent

* Include all E71,E72 errors

* Test the failing case in CI

* Revert "Test the failing case in CI"

This reverts commit 110add0a70.

* Push to test!
This reverts commit f317532779.

* ok back to passing
This reverts commit ba5052685f.

* Prove that CI fails when formatting is incorrect.

* Fix formatting

* Remove duplicitous E117 rule

* Use flake8 config for precommit

---------

Co-authored-by: waifairer <waifairer@gmail.com>
 2023-07-24 11:19:58 -04:00
George Hotz
7151382364 Refactor load/store before tensor cores (#1193) 
* minor cleanups

* render_const

* now that's a nice refactor

* clean up vload/vstore

* clean up render_load

* debugs there

* dumb

* err, this?

* const float4

* what's failing

* bugfix

* statement includes semicolon

* bugfix
 2023-07-08 15:54:58 -07:00
George Hotz
89b8b39d9c fix mypy 2023-05-13 21:25:36 -07:00
George Hotz
81aa3e546b exclude GPU on tiny (#766) 2023-05-05 10:07:23 -07:00
George Hotz
4f6d674ec0 use CPU tests in pre-commit 2023-05-03 19:46:16 +00:00