Skip to content

Commit

Permalink
Update comment for swizzleSharedMemory
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobhinkle committed Sep 5, 2024
1 parent ecefe56 commit f0f6702
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions csrc/scheduler/multi_matmul.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -60,12 +60,7 @@ inline void checkConcreteStaticDim(const AbstractId& abs_id) {
//! for matmul mainloop and epilogue.
//! The shared mem data layout is always 2D currently, and this utility
//! function assumes that the shared_mem_tv has the following structure:
//! [tile_row, tile_col, ***skip***] where the parameter `skip` is the number
//! of reduction domains to be skipped. The IDs of tile_row and tile_col are
//! the ones being swizzled.
//! If the input tensorview is not stored in shared memory, the function will
//! skip the actual swizzle. This is used to help the domain mapping between
//! mma_result and the epilogue tensor.
//! [tile_row, tile_col]
//! Returns the domain with swizzle. For the case of legacy swizzle, this
//! domain must be set as loop domain. For the case of new swizzle, this domain
//! must be set as allocation domain.
Expand Down

0 comments on commit f0f6702

Please sign in to comment.