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

[BACKEND] Solidify f8e4m3 (#2105)

Browse Source 
Co-authored-by: Philippe Tillet <phil@openai.com>
This commit is contained in:
Zahi Moudallal
2023-08-18 19:12:09 -07:00
committed by GitHub
parent 23ef2615d2
commit 23dd11d471
18 changed files with 171 additions and 172 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -245,7 +245,7 @@ class JITFunction(KernelInterface[T]):
dtype_str = str(key).split(".")[-1]
tys = {
"bool": "i1",
"float8e4": "fp8e4",
"float8e4nv": "fp8e4nv",
"float8e5": "fp8e5",
"float8e4b15": "fp8e4b15",
"float8e4b15x4": "fp8e4b15x4",