github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity
9,772 Commits 203 Branches 13 Tags
ca7a64144271744eb678a407a41f4d603f1e5944
Commit Graph

9772 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Sardor
ca7a641442 fix bugs at examples/yolov3.py (#11614) 
* Update load_weight. Give valid model url

* Fix bug in iou function
 2025-08-11 21:14:47 -04:00
chenyu
0c97d6de1b don't round pow output for int pow int (#11625) 
also added atol=0 and big pows for the tests
 2025-08-11 20:57:47 -04:00
chenyu
d623f6d850 support int Tensor pow to const non-negative int (#11624) 
matches torch
 2025-08-11 19:50:19 -04:00
chenyu
857a830dcc fix test_arange_float_step (#11623) 2025-08-11 16:58:42 -04:00
chenyu
0806677b51 rewrite sort idx (#11613) 2025-08-11 16:20:56 -04:00
George Hotz
700c11597b switch contextvars.ContextVar to _ContextVar (#11621) 2025-08-11 12:20:09 -07:00
ttomsa
ae0c3cfff6 change clang -march flag to -mcpu on arm (#10970) 
Co-authored-by: wozeparrot <wozeparrot@gmail.com>
 2025-08-11 13:38:48 -04:00
geohotstan
27bcb9fd1c Support cubic mode for ONNX Resize OP (#11612) 
* start

* add reference

* this is so much slower

* this makes sense but differs from official impl, but results are still correct..?

* add a comment

* Just keep it simple for now since I don't fully get it yet

* address comments

* correct

* teeny clean up

* another small comment improvement lol
 2025-08-11 11:49:30 -04:00
nimlgen
d2bb1bcb97 cloud: a bit better err handling (#11616) 
* cloud: err propagation to client

* fix

* print exc

* linter

* excs

* fix

* hm

* flaky
 2025-08-11 15:51:22 +03:00
qazal
6a232ccdac viz: add tiny range drawing helper (#11620) 
* viz: add tiny range drawing helper

* less
 2025-08-11 15:15:43 +03:00
qazal
e768773e13 viz: use colors helper (#11618) 2025-08-11 13:10:15 +03:00
qazal
7d6c0a8cc7 viz: refactor progress msg (#11617) 2025-08-11 13:01:36 +03:00
chenyu
630edcffd8 remove .float calls in olmoe (#11610) 
still matches torch
 2025-08-10 20:33:22 -04:00
chenyu
a67e0917c3 list indexing can normalize in python (#11609) 
* list indexing can normalize in python

list index does not need to be normalized in tensor

* update those
 2025-08-10 20:02:38 -04:00
chenyu
1181ec0cd2 few more tensor indexing test cases (#11608) 2025-08-10 18:56:42 -04:00
George Hotz
996c907c0b rewrite not ready + children machinery (#11607) 
* rewrite not ready + children machinery

* it doesn't like track rewrites
 2025-08-10 15:28:30 -07:00
Sieds Lykles
1875bc69f9 Late rewrite rules for CMPLT (#11591) 
* add rules

* more rules

* fix comment spelling

* remove two rules
 2025-08-10 22:18:13 +02:00
nimlgen
5403a4aeaf null dev: support offset on buffers (#11606) 
* null dev: support offset on buffers

* nolimit
 2025-08-10 21:58:37 +03:00
geohotstan
b0dab6a4cd onnx Resize OP clean up (#11603) 
* start

* slight clean up
 2025-08-10 14:10:39 -04:00
Sieds Lykles
10540414cd Add Ops.CMPEQ (#10431) 
* Add op

* add to Groupop.ALU

* fix spec

* fix ptx

* temporary pickle by name to see process replay

* add Ops.EQ to binary ops

* Actuall rename properly

* add test to assert CMPEQ is being used

* Ops.CMPEQ is automatic cast to bool

* add Ops.CMPEQ to llvm

* add Ops.CMPEQ to llvm
 2025-08-10 13:13:16 +02:00
chenyu
f7aa1b85fe minor sort cleanups (#11602) 2025-08-10 01:51:23 -04:00
chenyu
dfb702ef33 fix sort for small dim (#11601) 
* fix sort for small dim

* fixed test_sort_empty
 2025-08-10 01:17:41 -04:00
chenyu
ef17af85c6 remove .float call in llama logit (#11598) 
* remove .float call in llama logit

* bfloat item
 2025-08-10 00:02:18 -04:00
chenyu
dd3d2eb36c add training llama3 test in ci (#11599) 2025-08-09 22:35:39 -04:00
chenyu
3e64467322 remove freqs_cis contiguous in llama (#11597) 2025-08-09 21:11:12 -04:00
chenyu
7338ffead0 small beautiful_mnist update (#11596) 
gather is fast now. there's a conv/bw kernel that only gets fast with BEAM, but whole thing runs < 5 seconds now regardless
 2025-08-09 19:51:14 -04:00
chenyu
45baec1aab model parallel llama (#11588) 
MP=8 GRADIENT_ACC_STEPS=3 BS=1 DEFAULT_FLOAT=bfloat16 OPTIM_DTYPE=bfloat16 LLAMA3_SIZE=70B SEQLEN=512 PYTHONPATH=. MODEL=llama3 python3 examples/mlperf/model_train.py
 2025-08-09 16:54:27 -04:00
nimlgen
09bc377da3 search: print runtime failures on debug (#11593) 2025-08-09 23:01:19 +03:00
nimlgen
14f99ff1a1 amd: doorbell_cpu_addr is not used (#11592) 
* amd: doorbell_cpu_addr is not used

* hm
 2025-08-09 20:03:21 +03:00
Sieds Lykles
01c770c77b Fix z3 float cast in indexing (#11590) 
* adjust dtype of z3_renderer and add rule for cast

* dtypes.bool is also cast noop

* add regression test

* make embedding smaller

* even smaller test
 2025-08-09 17:59:23 +02:00
Sieds Lykles
10d388499d Refactor optional.py (#11578) 
* move fast_idiv to transcendental

* move optional.py

* adjust comment

* change import

* mypy needs this?
 2025-08-09 17:35:05 +02:00
nimlgen
20e46a175c do not use disk with usb (#11119) 
* not use disk with usb

* better name
 2025-08-09 11:58:02 +03:00
qazal
53179953fc viz: factor out memory graph render (#11586) 2025-08-08 20:18:11 +03:00
qazal
8ce72d3fad simpler disassembly table spec (#11583) 
* simpler disassembly table spec

* update ui

* move to scalar/vec render
 2025-08-08 17:59:26 +03:00
qazal
44a222a9b2 viz: move resource usage summary to server (#11582) 2025-08-08 17:08:28 +03:00
qazal
793ace530e update amd_uop_matmul.py import (#11581) 
Using this for testing SQTT
 2025-08-08 17:07:35 +03:00
chenyu
b232c60def benchmark openpilot 0.9.9 (#11575) 
* benchmark openpilot 0.9.9

not sure what to do with the 0.9.7 ones with IMAGE=2 and validate

* name
 2025-08-08 01:26:14 -04:00
qazal
16f0edbe90 pass opts arg in get_program process replay [pr] (#11571) 
* fix ptx process replay

* keyword arg

* renderer is also optional [pr]

* test_linearizer fixup

* name function order is args,ret,kwargs

* can use opts_to_apply

* pass through p.applied_opts

* sink_arg

* now it opens devices too
 2025-08-08 03:05:09 +03:00
qazal
960cc6533a pass through name function args in track_rewrites (#11572) 2025-08-08 02:28:52 +03:00
wozeparrot
1826004ef9 feat: add tinyos builder link (#11570) 2025-08-07 17:42:18 -04:00
George Hotz
82be8abfd2 move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
chenyu
702e38dc19 remove FUSE_ARANGE_UINT (#11567) 
also add IGNORE_OOB=1 to bert runs. lowered BS on tinybox to 90 since 96 oom during eval without reset
 2025-08-07 16:49:06 -04:00
George Hotz
6ed2dfd187 delete the arange dim mismatch restriction (#11568) 
* delete the arange dim mismatch restriction

* skip that test race
 2025-08-07 13:46:17 -07:00
wozeparrot
7ae4335127 feat: generate blend index (#11566) 2025-08-07 14:20:28 -04:00
chenyu
594cbdc66f skip AM ResNet50 benchmark (#11565) 
hanging with FUSE_ARANGE?
 2025-08-07 14:07:01 -04:00
chenyu
aa1a6f2132 support threshold in Tensor.softplus (#11564) 
fix gradient for large input
 2025-08-07 13:43:18 -04:00
chenyu
7ee3770961 FUSE_ARANGE=1 (#11427) 
* FUSE_ARANGE=1

* fix test

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-08-07 13:32:34 -04:00
George Hotz
4dfcfb1ae5 Revert "Revert "viz: align-center checkbox (#11555)"" 
This reverts commit c52facfd29.
 2025-08-07 08:15:57 -07:00
George Hotz
7e42427a7b Revert "Revert "viz: remove color for unbind step (#11554)"" 
This reverts commit 5650c7b86c.
 2025-08-07 08:15:51 -07:00
George Hotz
dc765fbeb7 Revert "viz: timeline perf (#11533)" 
This reverts commit 031f26632b.
 2025-08-07 08:08:51 -07:00