tinygrad

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-22 21:38:10 -05:00

Files

George Hotz b65b70812a Exec AST (#404 )

* working exec ast

* exec_ast is staticmethod

* GenericExecAST

* fold that sometimes

* ExplicitExecAST

* exec_ast for GPU

* gpu working

* get_lazyop_shape

* now gpubuffer is ExplicitExecAST

* dedup

* add a type

* RESHAPE in opencl code

* fix linter

* that too for linter

* cleanups

* remove dead code

* GenericShape is less lines

* add ALLOWED_KERNEL_COUNT to tests

* fix mypy

* that's gotta be recursive

* fix opencl shape processing

* remove unneeded lambda

2022-10-28 08:27:03 -07:00

ane

ane: procPath issue. don't waste more time with this, focus on core tinygrad

2022-08-16 10:36:13 +02:00

cherry

only resnet18, it's too slow otherwise

2021-10-30 16:48:39 -07:00

cuda

Accel/cuda (#319 )

2022-05-14 21:25:30 -07:00

llvm

Exec AST (#404 )

2022-10-28 08:27:03 -07:00

metal

a bit of work on metal

2021-12-30 13:53:08 -05:00

opencl

Exec AST (#404 )

2022-10-28 08:27:03 -07:00

rawcpu

tinygrad.nn (#367 )

2022-08-18 07:41:00 -07:00

tpu

header

2021-10-30 16:41:05 -07:00

MAPPING

fix ane on new mac os x

2022-08-06 19:10:22 +00:00

README

refactor efficientnet loading

2021-10-30 17:02:17 -07:00

README

This is where we scope out adding accelerators to tinygrad

ane -- Apple Neural Engine, in the M1 + newer iPhones
cherry -- Largely defunct custom hardware based on a RISC-V extension
tpu -- Google's TPU, available for rent in Google Cloud