sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions Activity
11,925 Commits 336 Branches 15 Tags
amd_refac
Commit Graph
11925 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
geohot 113e15a81f line 2026-01-28 17:11:04 +08:00
geohot 5987f3a9b6 kimi cleanups 2026-01-28 17:01:14 +08:00
geohot 86439561fa a lil more 2026-01-28 13:29:27 +08:00
geohot c49aca3bc0 wmma refactor 2026-01-28 13:24:56 +08:00
geohot 0afa914063 Fmt matters it seems 2026-01-28 13:14:14 +08:00
geohot 5c33ff6536 correct fix + tests 2026-01-28 12:48:42 +08:00
geohot 8eaa6b593e fix tests 2026-01-28 12:40:49 +08:00
geohot cd12f54982 smem test 2026-01-28 12:35:40 +08:00
geohot 4ae8cf9184 junk remove 2026-01-28 12:24:48 +08:00
geohot d697f7073a bits junk remove 2026-01-28 12:21:23 +08:00
geohot 09709b082a remove bad helpers 2026-01-28 12:17:01 +08:00
geohot 68abb5bf2d write clean code 2026-01-28 12:07:00 +08:00
geohot 313897eb2e dup code 2026-01-28 12:04:38 +08:00
geohot 8e1fbe2f5b use the correct apis 2026-01-28 12:00:47 +08:00
geohot 2a9e0af9b4 no ctx 2026-01-28 11:51:36 +08:00
geohot d7b2033009 flat 2026-01-28 11:46:07 +08:00
geohot ef82390499 merge 2026-01-28 11:23:49 +08:00
geohot 6d45afd1ec simpler 2026-01-28 11:19:19 +08:00
George HotzandGitHub 88bc5ee212 assembly/amd: rename to better names (#14384) 
* assembly/amd: rename to better names

* might help fuzzing segfault

* emu2 -> emu
 2026-01-28 10:00:54 +08:00
George HotzandGitHub 065b95cfb0 Revert "add retry to fetch (#14370)" (#14385) 
This reverts commit dc4d7f2d55.
 2026-01-28 09:35:37 +08:00
Eitan TurokandGitHub dc4d7f2d55 add retry to fetch (#14370) 2026-01-27 14:04:25 -08:00
chenyuandGitHub 8d1f3c8885 fix copysign for inf input (#14381) 
* fix copysign for inf input

* llvm olt
 2026-01-27 16:45:48 -05:00
sirhcmandGitHub 289a3e415e also skip test_nonoverlapping_shrink_assignment (#14382) 2026-01-27 16:26:26 -05:00
sirhcmandGitHub f34efc1ad1 DISABLE_FAST_IDIV actually works as a ContextVar (#14378) 2026-01-27 16:12:42 -05:00
chenyuandGitHub 8c899e4aaf fix copysign for -0 (#14380) 
test both x and 1/x < 0 work too. and found another big with the * 0 hack
 2026-01-27 15:44:58 -05:00
chenyuandGitHub 62884585a7 failed test case for copysign -0.0 (#14379) 
* failed test case for copysign -0.0

* skip those
 2026-01-27 14:37:17 -05:00
nimlgenandGitHub ec1b28bc2c am: exit early in case of failures (#14376) 
* am: exit early in case of failures

* sorry, pre-linter

* reset when error state
 2026-01-27 22:10:02 +03:00
chenyuandGitHub cd22ee9ed0 add InvalidType to ConstType [pr] (#14373) 
* add InvalidType to ConstType [pr]

TYPED=1 python test/test_tiny.py passes.
added PyConst = float|int|bool for some Tensor level input types

* hcq
 2026-01-27 14:09:34 -05:00
sirhcmandGitHub 5b42a1357b SCACHE=0 works with DEBUG (#14377) 2026-01-27 13:12:43 -05:00
chenyuandGitHub db010a31be IGNORE_OOB -> CHECK_OOB [pr] (#14374) 
flip the meaning
 2026-01-27 12:20:59 -05:00
chenyuandGitHub c22667b0c4 also skip test_overlapping_shrink_assignment_reverse (#14375) 
crashing
 2026-01-27 12:20:39 -05:00
nimlgenandGitHub e52d58b041 autogen: update amd (#14372) 2026-01-27 19:53:14 +03:00
nimlgenandGitHub cbf94a0a95 nv: exit early in case of failures (#14363) 
* nv: exit early in case of failures

* f

* cleaner
 2026-01-27 19:16:22 +03:00
nimlgenandGitHub ec691cb299 am: print sq intrs (#14366) 
* am: print sq intrs

* cleaner
 2026-01-27 18:28:13 +03:00
qazalandGitHub a5f3d46423 hcq: do not assume kernel names are unique (#14371) 
* hcq: do not assume kernel names are unique

* colored kernel name
 2026-01-27 23:03:15 +09:00
George HotzandGitHub e5df7e640b fix branches in amd_asm_matmul (#14369) 2026-01-27 20:48:42 +08:00
geohot 0ced258726 HOTFIX: skip crashing assign test 2026-01-27 20:35:17 +08:00
George HotzandGitHub 131ae604de force_transcendental on sqrt (#14368) 2026-01-27 20:24:41 +08:00
imaoloGitHubGeorge Hotz
 		14574c68fa Add ContextVar to disable the scheduler cache (#14257) 
* add scheduler cache ContextVar

* test scheduler cache context var

---------

Co-authored-by: George Hotz <[email protected]>
 2026-01-27 19:55:29 +08:00
George HotzandGitHub bfc88bcfb8 assembly/amd: emu refactors + enable PYTHON_REMU by default (#14361) 
* assembly/amd: start refactors

* cleanups

* those are global

* methods on ctx

* const cleanup

* range helper

* types and imports

* cleanups

* cleanups

* remove stale name

* fix emu2 types

* more typing

* more mypy

* cleanups

* fxns

* scc cleanup

* cleanups

* cleanups

* simpler parse_pcode

* laneid

* no defaults for pcode

* pcode is not optional

* cleanups

* functions cleanup

* splat

* expr_parser functions

* single tok

* invert global loops

* try_eat

* minor

* run parser on all

* no silent 0

* tests
 2026-01-27 17:42:24 +08:00
sirhcmandGitHub 2e72625652 Revert "decompose dtypes.long to ints where unsupported (#14261)" (#14362) 2026-01-27 02:04:59 -05:00
qazalandGitHub f866b2a513 mfma loop in asm dsl (#14349) 
* mfma loop in asm dsl

* work
 2026-01-27 11:11:37 +09:00
sirhcmandGitHub 0793319929 decompose dtypes.long to ints where unsupported (#14261) 
* add works

* use carry not overflow

* bitwise ops

* use tag instead of vec

* cleaner

* mul somewhat works

* mul actually works

* SUB and NEG work

* SHL/SHR

* ulong support

* this should work?

* oops

* fix indexing

* all ALU mostly works

* refactor

* test_dtype passing

* signed division works

* format

* clean

* some tests

* ruff
 2026-01-26 18:34:13 -05:00
wozeparrotandGitHub a987a4abc3 feat: llama8b dev_beam.sh (#14358) 2026-01-26 14:51:23 -08:00
sirhcmandGitHub c9c533fc78 libclang path is homebrew on macos (#14357) 
* libclang path is homebrew macos

* typo

* ugh

* typo

* regen

* no LIBCLANG_PATH
 2026-01-26 17:32:09 -05:00
chenyuandGitHub d641e63189 improve min/max for AND (#14356) 2026-01-26 15:44:18 -05:00
chenyuandGitHub f16372487a fix assign hazard on shrink (#14355) 
* fix assign hazard on shrink

possible to have race if both assign src and dest are shrink

* test_nonoverlapping_shrink_assignment
 2026-01-26 14:46:30 -05:00
chenyuandGitHub 145df879c1 find_permutes -> fix_assign_hazard [pr] (#14354) 
some noop tweaks and comment updates
 2026-01-26 14:05:19 -05:00
nimlgenandGitHub e152f1b0f5 llama: use ALL2ALL (#14353) 2026-01-26 22:01:53 +03:00
nimlgenandGitHub 3f25eb3026 am: ih (#14346) 
* am: ih

* um

* fix

* line

* no trap and fix ring

* keep

* fix
 2026-01-26 20:11:04 +03:00