[FRONTEND] Add PyTorch fp8 dtypes to Triton (triton-lang#2279)

Add PyTorch fp8 dtypes (https://github.com/pytorch/pytorch/blob/8025b193a966a6d8e3afc9c03a54e577bc04eb3d/torchgen/api/types/types.py#L50-L51) to Triton.
BinFan · Sep 12, 2023 · 37f1249 · 37f1249
1 parent a47f1f5
commit 37f1249
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/triton/runtime/jit.py b/python/triton/runtime/jit.py
@@ -250,6 +250,8 @@ def _type_of(key):
             "float8e5": "fp8e5",
             "float8e4b15": "fp8e4b15",
             "float8e4b15x4": "fp8e4b15x4",
+            "float8_e4m3fn": "fp8e4nv",
+            "float8_e5m2": "fp8e5",
             "float16": "fp16",
             "bfloat16": "bf16",
             "float32": "fp32",