From 2bfc6c4bdef3a8b33bb8b53fae5b82e44df480f0 Mon Sep 17 00:00:00 2001 From: Zhongdi LUO Date: Fri, 10 Jul 2026 13:01:58 +0000 Subject: [PATCH] fix: pack scalar tmem softmax as fp16 --- hw/rtl/core/VX_execute.sv | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hw/rtl/core/VX_execute.sv b/hw/rtl/core/VX_execute.sv index 86e3d9ed..5077bd25 100644 --- a/hw/rtl/core/VX_execute.sv +++ b/hw/rtl/core/VX_execute.sv @@ -418,7 +418,7 @@ module VX_execute import VX_gpu_pkg::*; #( input [31:0] denom_bits; begin `ifdef SV_DPI - scalar_tmem_softmax_prob_pack = 32'(dpi_softmax_prob_to_fp8e4m3x4(1'b1, int'(score_bits), int'(max_bits), int'(denom_bits))); + scalar_tmem_softmax_prob_pack = 32'(dpi_softmax_prob_to_f16x2(1'b1, int'(score_bits), int'(max_bits), int'(denom_bits))); `else scalar_tmem_softmax_prob_pack = '0; `endif