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
|
34eaab4c87
|
sgemm_tcore: Fix warp-specialized kernel for larger dim
|
2024-06-11 20:50:32 -07:00 |
|
Hansung Kim
|
39449ece37
|
Add warp-specialized{, and dma enabled} kernel
NOTE: warpspecial_dma is hacked, need to get rid of dma invocation in
the consmer code.
|
2024-06-10 16:39:49 -07:00 |
|