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

fix: update fp16 softmax case25 score expectation

Browse Source
This commit is contained in:
Zhongdi LUO
2026-07-11 00:36:25 +00:00
parent ca42426e5c
commit ec437cce6e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kernels/wu_arch_cases/case25_scalar_tmem_softmax_vector/kernel.cpp
View File

@@ -6,7 +6,7 @@
#define WU_CASE25_P_READY_BASE 0xb600u #define WU_CASE25_P_READY_BASE 0xb600u
#define WU_CASE25_DONE_BASE 0xb700u #define WU_CASE25_DONE_BASE 0xb700u
#define WU_CASE25_FP32_ZERO 0x00000000u #define WU_CASE25_FP32_ZERO 0x00000000u
#define WU_CASE25_FP32_THIRTY_SIX 0x42100000u #define WU_CASE25_FP32_THIRTY_TWO 0x42000000u
extern "C" { extern "C" {
volatile uint32_t g_case25_q_row[4] __attribute__((aligned(16))) = { volatile uint32_t g_case25_q_row[4] __attribute__((aligned(16))) = {
@@ -145,7 +145,7 @@ extern "C" int wu_main() {
asm volatile("fence rw, rw" ::: "memory"); asm volatile("fence rw, rw" ::: "memory");
if (tid == 0 && g_case_mem[1] == 0 && if (tid == 0 && g_case_mem[1] == 0 &&
observed != WU_CASE25_FP32_THIRTY_SIX) { observed != WU_CASE25_FP32_THIRTY_TWO) {
g_aux[0] = observed; g_aux[0] = observed;
g_case_mem[1] = 0x92u; g_case_mem[1] = 0x92u;
} }