Fix truncated bits for boolean in swizzling path

Signed-off-by: Witold Dziurdz <[email protected]>

Author: wdziurdz

lib/Conversion/TritonGPUToLLVM/ConvertLayoutOpToLLVM.cpp

@@ -144,6 +144,14 @@ struct ConvertLayoutOpUsingLinearLayoutsConversion
           inVals, [&](Value v) { return b.zext(i8ElemTy, v).getResult(); }));
       auto outVals = transferWithinBlockSwizzlingImpl(
           loc, rewriter, srcLayout, dstLayout, newInVals, i8ElemTy, smemBase);
+      if (llvmElemTy.getIntOrFloatBitWidth() == 1) {
+        auto zero = rewriter.create<LLVM::ConstantOp>(
+            loc, i8_ty, rewriter.getI8IntegerAttr(0));
+        for (auto &v : outVals) {
+          v = b.icmp_ne(v, zero);
+        }
+        return outVals;
+      }
       for (auto &v : outVals) {
         v = b.trunc(llvmElemTy, v);
       }