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

Fix potential bitwidth bug in compute API

Browse Source
This commit is contained in:
Hansung Kim
2024-11-09 20:59:58 -08:00
parent 68054689c9
commit cb916ead39
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
kernel/include/gemmini_mmio.h
View File

@@ -114,8 +114,9 @@ inline void gemmini_tile_compute(const uint32_t a_hexadecile,
const uint32_t d_hexadecile, const uint32_t d_hexadecile,
const bool accumulate) { const bool accumulate) {
if constexpr (!store_to_spad) { if constexpr (!store_to_spad) {
GEMMINI_CISC_CMD_R((accumulate << 24) | (b_hexadecile << 16) | GEMMINI_CISC_CMD_R((static_cast<uint32_t>(accumulate) << 24) |
(a_hexadecile << 8) | GEMMINI_CISC_COMPUTE_HEXADECILES); (b_hexadecile << 16) | (a_hexadecile << 8) |
GEMMINI_CISC_COMPUTE_HEXADECILES);
} else { } else {
GEMMINI_CISC_CMD_R((d_hexadecile << 24) | (b_hexadecile << 16) | GEMMINI_CISC_CMD_R((d_hexadecile << 24) | (b_hexadecile << 16) |
(a_hexadecile << 8) | GEMMINI_CISC_COMPUTE_AND_STORE_TO_SPAD); (a_hexadecile << 8) | GEMMINI_CISC_COMPUTE_AND_STORE_TO_SPAD);