Yoshinori SanoandGitHub edb74897b2 support safe load bf16 (#3310) ...

* support safe load bf16

* fix lint error E501

* add test for loading safetensors

* key should be BOOL

* fix lint

2024-02-04 10:08:39 -05:00

..

codegen

wmma: add HIP FP16 to FP16 tensor core (#3287)

2024-01-31 23:00:51 -05:00

features

limit group_for_reduce bufs to 32kb (#3299)

2024-02-02 03:13:12 -05:00

nn

support safe load bf16 (#3310)

2024-02-04 10:08:39 -05:00

renderer

wmma: add HIP FP16 to FP16 tensor core (#3287)

2024-01-31 23:00:51 -05:00

runtime

clean up einsum_mulacc (#3312)

2024-02-04 06:21:19 -05:00

shape

touchup canonicalize empty mask (#3308)

2024-02-03 21:05:10 -05:00

__init__.py

move dtypes to dtype.py (#2964)

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

device.py

type annotation for Compiler.cachekey and minor cleanup (#3298)

2024-02-01 21:31:21 -05:00

dtype.py

const cleanup with dtype.Scalar (#3257)

2024-01-26 21:16:22 -05:00

graph.py

remove unused expr node (#3170)

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

helpers.py

hip mutex signal (#3234)

2024-01-24 13:23:09 -08:00

jit.py

fix jit realize issue (#3258)

2024-01-26 18:27:35 -08:00

lazy.py

fix pad 0 size (#3277)

2024-01-30 08:58:10 -08:00

mlops.py

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

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

ops.py

hip events work (#3229)

2024-01-24 11:49:53 -08:00

realize.py

minor hip cleanups (#3237)

2024-01-24 15:13:38 -08:00

tensor.py

remove some contiguous and contiguous_backward from wino (#3306)

2024-02-02 23:15:05 -05:00