Commit Graph

6 Commits

Author SHA1 Message Date
Hansung Kim
ce4f3a24e3 sgemm_tcore: Replace hardcoded NUM_LANES with NUM_THREADS 2024-06-12 21:01:37 -07:00
Hansung Kim
635da96154 sgemm_tcore: Constify smem pointer for wmma_load 2024-06-12 13:36:29 -07:00
Hansung Kim
32e31c51a4 sgemm_tcore: Blocksize 64; Fix kernel launch on larger dim
& fix addrgen assembly too large offset error
2024-06-11 22:27:12 -07:00
Hansung Kim
03d1df8f53 sgemm_tcore: Separate transpose control on AS read/write
Make separate control flags on transposed AS read/write to make it easy
to model bank-conflict-free GMEM _and_ SMEM access.
2024-06-11 21:16:23 -07:00
Hansung Kim
9febfb9bdc sgemm_tcore: Move global_dmem_load back to kernel.cpp 2024-06-11 20:12:30 -07:00
Hansung Kim
ca7fd84a83 sgemm_tcore: Split util functions to a header file 2024-06-11 19:06:22 -07:00