diff --git a/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp index bdc58e1f9..9c954929d 100644 --- a/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp +++ b/lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp @@ -4091,7 +4091,7 @@ struct InsertSliceAsyncOpConversion rewriter, loc, this->getTypeConverter()->getIndexType(), i); v = insert_element(vecTy, v, falseVal, indexVal); } - v = bitcast(v, IntegerType::get(getContext(), byteWidth)); + v = bitcast(v, IntegerType::get(getContext(), bitWidth)); // Write shared memory if predicate is true PTXBuilder ptxStoreBuilder; auto *valOperand = ptxStoreBuilder.newOperand(v, "r");