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

Turn off TENSOR_HOPPER for Virgo flash

Browse Source
This commit is contained in:
Hansung Kim
2025-01-30 01:22:50 -08:00
parent 4f12227327
commit 96500e0abc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kernels/sgemm_tcore/sgemm_impl.hpp
View File

@@ -19,7 +19,7 @@ using float_type = float16_t;
// Generate kernel for the Hopper-style SMEM-decoupled tensor core. This uses // Generate kernel for the Hopper-style SMEM-decoupled tensor core. This uses
// asynchronous HGMMA and HGMMA_WAIT instructions. // asynchronous HGMMA and HGMMA_WAIT instructions.
#define TENSOR_HOPPER 1 #define TENSOR_HOPPER 0
// Constraints on parameters: // Constraints on parameters:
// * Memory: // * Memory: