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

sgemm_tcore: Double-buffer over K-dimension

Browse Source 
TODO: Not completely parameterized with DOUBLE_BUFFER yet.
This commit is contained in:
Hansung Kim
2024-06-04 18:23:27 -07:00
parent 18e3653d31
commit d8944db369
2 changed files with 155 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/regression/sgemm_tcore/main.cpp
View File

@@ -155,9 +155,9 @@ int main(int argc, char *argv[]) {
RT_CHECK(vx_dev_open(&device));
// FIXME: hardcoded
uint32_t dim_m = 16;
uint32_t dim_n = 16;
uint32_t dim_k = 16;
uint32_t dim_m = 32;
uint32_t dim_n = 32;
uint32_t dim_k = 32;
generate_source_matrix(dim_m, dim_n, dim_k);
generate_reference_matmul(dim_m, dim_n, dim_k);