chenyuandGitHub dcf7ecaaff update jit type annotation post lazy rewrite (#3091)

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

..

codegen

use device from LinearizerOptions in kernel search (#3090)

2024-01-11 14:46:03 -05:00

features

use device from LinearizerOptions in kernel search (#3090)

2024-01-11 14:46:03 -05:00

nn

make Embedding device aware for multigpu (#3051)

2024-01-08 20:09:26 -08:00

renderer

wmma: clean up device specific tensor core code (#3081)

2024-01-10 21:03:09 -05:00

runtime

use device from LinearizerOptions in kernel search (#3090)

2024-01-11 14:46:03 -05:00

shape

minor cleanup of View.reshape (#3088)

2024-01-11 13:05:54 -05:00

__init__.py

move dtypes to dtype.py (#2964)

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

device.py

use device from LinearizerOptions in kernel search (#3090)

2024-01-11 14:46:03 -05:00

dtype.py

pass tuple for strs to startswith (#2986)

2024-01-02 19:51:15 -05:00

graph.py

early gate the graph (#3070)

2024-01-09 20:17:13 -08:00

helpers.py

mem_estimate is always int, not symbolic (#3083)

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

jit.py

update jit type annotation post lazy rewrite (#3091)

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

lazy.py

minor cleanups of lazy.py (#3080)

2024-01-10 20:17:56 -05:00

mlops.py

grad -> grad_output in mlops for consistency (#2982)

2024-01-02 18:03:55 -05: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

Tensor.expand resolves the new_shape before shortcut return (#3078)

2024-01-10 14:29:15 -05:00