John Smith 44978669cf Add gradient checkpointing		2023-03-23 08:25:29 +00:00
..
autograd_4bit.py	add more scripts and adjust code for transformer branch	2023-03-22 04:09:04 +00:00
gradient_checkpointing.py	Add gradient checkpointing	2023-03-23 08:25:29 +00:00
quant_cuda.cpp	add fast_4bit_matmul and auto switch 2 methods according to bottleneck	2023-03-21 08:43:07 +00:00
quant_cuda_kernel.cu	add fast_4bit_matmul and auto switch 2 methods according to bottleneck	2023-03-21 08:43:07 +00:00