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

Revert "[FRONTEND][BACKEND] improved fp8 specs (#1841)" (#1865)

Browse Source 
This reverts commit d4c941177e.
This commit is contained in:
Philippe Tillet
2023-06-29 18:07:01 -07:00
committed by GitHub
parent 3be060849a
commit f77015967d
13 changed files with 550 additions and 415 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -208,9 +208,8 @@ class JITFunction(KernelInterface[T]):
dtype_str = str(key).split(".")[-1]
tys = {
"bool": "i1",
"float8e4": "fp8e4",
"float8e5": "fp8e5",
"float8e4b15": "fp8e4b15",
"float8e4": "fp8e4",
"float16": "fp16",
"bfloat16": "bf16",
"float32": "fp32",