fix: pack scalar tmem softmax as fp16

2026-07-10 13:01:58 +00:00
1 changed files with 1 additions and 1 deletions
--- 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