Tri Dao 78b7a1dc18 [OPT] Load fp16 weights on CPU before moving to GPU		2023-01-22 17:01:32 -08:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
bert.py	[FusedDense] Support relu, rename FusedDenseGeluDense -> FusedMLP	2023-01-17 18:12:27 -08:00
gpt.py	[OPT] Load fp16 weights on CPU before moving to GPU	2023-01-22 17:01:32 -08:00
opt.py	[OPT] Load fp16 weights on CPU before moving to GPU	2023-01-22 17:01:32 -08:00
vit.py	[FusedDense] Support relu, rename FusedDenseGeluDense -> FusedMLP	2023-01-17 18:12:27 -08:00