kooshi 2bc64597aa model parallelism

2023-03-24 23:03:43 -05:00

..

autograd_4bit.py

model parallelism

2023-03-24 23:03:43 -05:00

gradient_checkpointing.py

Add gradient checkpointing

2023-03-23 08:25:29 +00:00

quant_cuda_kernel.cu

fix bug

2023-03-23 23:37:39 +08:00

quant_cuda.cpp

add fast_4bit_matmul and auto switch 2 methods according to bottleneck

2023-03-21 08:43:07 +00:00