George HotzandGitHub cec0a7bc37 use shard api to eval resnet fast (#3136) ...

* use shard api to eval resnet fast

* to supports shard

* test to in multitensor

2024-01-15 16:49:38 -08:00

..

codegen

wmma: enable METAL half tensor cores and clean up cstyle (#3095)

2024-01-12 16:25:28 -05:00

features

catch runtime error in search._time_program (#3106)

2024-01-12 21:53:13 -05:00

nn

minor cleanup of test_disk_tensor (#3112)

2024-01-13 20:54:58 -05:00

renderer

wmma: enable METAL half tensor cores and clean up cstyle (#3095)

2024-01-12 16:25:28 -05:00

runtime

Christopherm99 ptx (#3139)

2024-01-15 16:44:20 -08:00

shape

update NumNode.__hash__ to be hash(self.b) (#3105)

2024-01-12 19:46:21 -05:00

__init__.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

device.py

fast resnet eval (#3135)

2024-01-15 14:15:18 -08:00

dtype.py

simplify dtype (#3137)

2024-01-15 16:27:43 -08:00

graph.py

s/lazydata/lazyop/ in print_tree (#3138)

2024-01-15 19:38:27 -05:00

helpers.py

fast resnet eval (#3135)

2024-01-15 14:15:18 -08:00

jit.py

update jit type annotation post lazy rewrite (#3091)

2024-01-11 15:49:30 -05:00

lazy.py

move fromcpu out of lazy.py (#3122)

2024-01-14 18:21:08 -08:00

mlops.py

Fix backward fn for < and == (#3037)

2024-01-14 20:39:52 -08:00

ops.py

remove unused reciprocal (#3053)

2024-01-09 08:59:04 -08:00

realize.py

mem_estimate is always int, not symbolic (#3083)

2024-01-10 23:39:51 -05:00

tensor.py

use shard api to eval resnet fast (#3136)

2024-01-15 16:49:38 -08:00