github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Add PyTorch fp8 dtypes
(8025b193a9/torchgen/api/types/types.py (L50-L51))
to Triton.
This commit is contained in:
Ying Zhang
2023-09-12 08:57:01 -07:00
committed by GitHub
parent a47f1f5c28
commit 37f12497b0
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/runtime/jit.py
View File

@@ -250,6 +250,8 @@ class JITFunction(KernelInterface[T]):
"float8e5": "fp8e5",
"float8e4b15": "fp8e4b15",
"float8e4b15x4": "fp8e4b15x4",
"float8_e4m3fn": "fp8e4nv",
"float8_e5m2": "fp8e5",
"float16": "fp16",
"bfloat16": "bf16",
"float32": "fp32",