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
nn	Fixed LinearFP8 and added tests.	2023-02-13 17:48:52 -08:00
optim
__init__.py	Added fp8 simulation layer.	2023-02-13 16:53:07 -08:00
__main__.py
cextension.py
functional.py	Added more blocksizes for stochastic rounding; fixed dequant blocksize.	2023-02-14 13:55:17 -08:00
utils.py