From 0e7f7d135667d51f7dd20fbb3ca1eb79e4daf198 Mon Sep 17 00:00:00 2001 From: Zahi Moudallal Date: Wed, 13 Sep 2023 10:02:34 -0700 Subject: [PATCH] Fixing assert --- lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVMBase.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVMBase.h b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVMBase.h index 78c4f92caa53..667f174f2270 100644 --- a/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVMBase.h +++ b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVMBase.h @@ -339,8 +339,9 @@ class ConvertTritonGPUOpToLLVMPatternBase { // Order auto inOrder = triton::gpu::getOrder(srcEncoding); auto outOrder = triton::gpu::getOrder(resSharedLayout); - assert(outVec * (maxPhase - 1) <= srcShape[outOrder[0]] && - "Swizzling would generate out of bounds memory accesses"); + assert(maxPhase == 1 || + outVec * maxPhase <= srcShape[outOrder[0]] && + "Swizzling would generate out of bounds memory accesses"); // Tensor indices held by the current thread, as LLVM values auto srcIndices = emitIndices(loc, rewriter, srcEncoding, srcTy, false); // Swizzling with leading offsets (e.g. Hopper GMMA)