Tim Dettmers f64cfe65aa Fixed prefetch bug for non-paged tensors; added benchmark.		2023-05-06 21:49:16 -07:00
..
autograd	Refactor FP4 into 4Bit and integrate NF4 data type.	2023-04-03 11:00:12 -07:00
cuda_setup	Added fp4 quant/dequant and dequant optimizations.	2023-02-04 14:52:04 -08:00
nn	CUTLASS compiles.	2023-04-25 17:15:51 -07:00
optim	Fixed prefetch bug for non-paged tensors; added benchmark.	2023-05-06 21:49:16 -07:00
__init__.py	Refactor FP4 into 4Bit and integrate NF4 data type.	2023-04-03 11:00:12 -07:00
__main__.py	Sort imports	2022-10-27 13:15:21 +02:00
cextension.py	Added paged optimizers.	2023-05-06 14:59:29 -07:00
functional.py	Changed prefetching.	2023-05-06 18:59:59 -07:00
utils.py	Some small changes.	2023-03-27 09:12:57 -07:00