wu-arch/vortex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: pack scalar tmem softmax as fp16

Browse Source
This commit is contained in:
Zhongdi LUO
2026-07-10 13:01:58 +00:00
parent dff1107bf5
commit 2bfc6c4bde
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
hw/rtl/core/VX_execute.sv
View File

@@ -418,7 +418,7 @@ module VX_execute import VX_gpu_pkg::*; #(
input [31:0] denom_bits; input [31:0] denom_bits;
begin begin
`ifdef SV_DPI `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 `else
scalar_tmem_softmax_prob_pack = '0; scalar_tmem_softmax_prob_pack = '0;
`endif `endif