Tim Dettmers 2489d819c5 Added more blocksizes for stochastic rounding; fixed dequant blocksize.		2023-02-14 13:55:17 -08:00
..
autograd	Added more blocksizes for stochastic rounding; fixed dequant blocksize.	2023-02-14 13:55:17 -08:00
cuda_setup	Added error message for unexpected CUDA exception.	2023-01-03 06:57:07 -08:00
nn	Fixed LinearFP8 and added tests.	2023-02-13 17:48:52 -08:00
optim	Fix several typos in logging and comments	2022-11-01 09:53:47 +01:00
__init__.py	Added fp8 simulation layer.	2023-02-13 16:53:07 -08:00
__main__.py	Sort imports	2022-10-27 13:15:21 +02:00
cextension.py	Fixed issue where the CUDA SETUP was not printed.	2023-01-04 03:50:53 -08:00
functional.py	Added more blocksizes for stochastic rounding; fixed dequant blocksize.	2023-02-14 13:55:17 -08:00
utils.py	Added outlier detector and fake quantization layer.	2023-01-28 17:05:22 -08:00