pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-11-11 22:34:53 +08:00

Files

Amadeusz Skrzypczak b64bd4a5dd Add torch.float8_e5m2 and torch.float8_e4m3 data types (#104242 )

Proposal of two float8 variants - e5m2 and e4m3 - based on https://arxiv.org/pdf/2209.05433.pdf

Hide all Float8 operator implementations behind `#if !defined(C10_MOBILE)` guard to keep Android build size almost unchanged

TODO:
 - Refactor duplicated code
 - Cleanup unbalanced pragma pop in dtype utils
 - Add native implementation on the CUDA size

Co-authored-by: Nikita Shulga <nshulga@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/104242
Approved by: https://github.com/albanD

2023-07-20 16:09:11 +00:00

all.imp

Add support for include-what-you-use (#71114 )

2022-02-04 01:39:48 +00:00

aten.imp

Add support for include-what-you-use (#71114 )

2022-02-04 01:39:48 +00:00

c10.imp

Add torch.float8_e5m2 and torch.float8_e4m3 data types (#104242 )

2023-07-20 16:09:11 +00:00

fixup.py

Apply ufmt linter to all py files under tools (#81285 )