aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorWon Jeon <won.jeon@arm.com>2024-04-17 18:06:22 +0000
committerEric Kunze <eric.kunze@arm.com>2024-04-18 21:29:13 +0000
commit7296e16351554a2c80daab18b2d69b44c67b807a (patch)
treecc26f697937f64fb4212b2ad25860cc07c40aeec
parent3eec59b22128462eef668dbfb770c6924aeee40f (diff)
downloadreference_model-7296e16351554a2c80daab18b2d69b44c67b807a.tar.gz
Remove invalid bit calculation from FP32 to FP8
This is a temporary change and waiting for compliance testing code for BF16/FP8. Signed-off-by: Won Jeon <won.jeon@arm.com> Change-Id: I61a548cffedb5f16947cd1029999ccd4ff25a535
-rw-r--r--verif/generator/tosa_utils.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py
index 478190d..d4f3898 100644
--- a/verif/generator/tosa_utils.py
+++ b/verif/generator/tosa_utils.py
@@ -278,16 +278,16 @@ def float32_to_bfloat16(f):
def float32_to_fp8e4m3(f):
"""Turns fp32 value into fp8e4m3"""
f32_bits = get_float32_bitstring(f)
- fp8_bits = f32_bits[0] + f32_bits[1:5] + f32_bits[9:12] + "0" * 24
- fp_bytes = int(fp8_bits, 2).to_bytes(4, byteorder=sys.byteorder)
+ # TODO: needs src/generate and src/verify code ready
+ fp_bytes = int(f32_bits, 2).to_bytes(4, byteorder=sys.byteorder)
return struct.unpack("@f", fp_bytes)[0] # native byteorder
def float32_to_fp8e5m2(f):
"""Turns fp32 value into fp8e5m2"""
f32_bits = get_float32_bitstring(f)
- fp8_bits = f32_bits[0] + f32_bits[1:6] + f32_bits[9:11] + "0" * 24
- fp_bytes = int(fp8_bits, 2).to_bytes(4, byteorder=sys.byteorder)
+ # TODO: needs src/generate and src/verify code ready
+ fp_bytes = int(f32_bits, 2).to_bytes(4, byteorder=sys.byteorder)
return struct.unpack("@f", fp_bytes)[0]