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/BACKEND] Revived Float8E4B15x4 (#2090)

Browse Source
This commit is contained in:
Zahi Moudallal
2023-08-11 17:49:52 -07:00
committed by GitHub
parent 382e8fb1fa
commit a01c116f76
12 changed files with 59 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -248,6 +248,7 @@ class JITFunction(KernelInterface[T]):
"float8e4": "fp8e4",
"float8e5": "fp8e5",
"float8e4b15": "fp8e4b15",
"float8e4b15x4": "fp8e4b15x4",
"float16": "fp16",
"bfloat16": "bf16",
"float32": "fp32",