github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Changed warpSize = 32 to warpSize = 64

Browse Source
This commit is contained in:
B1tway
2023-03-06 18:33:35 +00:00
parent f9bd9908a1
commit 693cf89777
10 changed files with 65 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Conversion/TritonGPUToLLVM/ConvertLayoutOpToLLVM.cpp
View File

@@ -93,7 +93,11 @@ private:
SmallVector<Value> mmaColIdx(4);
SmallVector<Value> mmaRowIdx(2);
Value threadId = getThreadId(rewriter, loc);
#ifdef USE_ROCM
Value warpSize = idx_val(64);
#else
Value warpSize = idx_val(32);
#endif
Value laneId = urem(threadId, warpSize);
Value warpId = udiv(threadId, warpSize);
// TODO: fix the bug in MMAEncodingAttr document