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

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_device.py

fix UnboundLocalError when running Compiler with DISABLE_COMPILER_CACHE (#3296)

2024-02-01 21:12:33 -05:00

test_disk_cache.py

fast resnet eval (#3135)

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

test_disk_tensor.py

support safe load bf16 (#3310)

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

test_flopcounter.py

move dtypes to dtype.py (#2964)

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

test_helpers.py

move TestGetContraction to test_helpers (#3313)

2024-02-04 06:05:01 -05:00

test_shapetracker_math.py

remove unused expr node (#3170)

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

test_shapetracker.py

move TestGetContraction to test_helpers (#3313)

2024-02-04 06:05:01 -05:00

test_shm_tensor.py

ops_shm removed (#2351)

2023-11-18 11:41:58 -08:00

test_symbolic.py

name upcast variables (#3200)

2024-01-22 11:37:28 -05:00