diff --git a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td --- a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td +++ b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td @@ -296,8 +296,9 @@ OptionalAttr:$alignment); let results = (outs Res]>:$res); string llvmBuilder = [{ + auto addrSpace = $_resultType->getPointerAddressSpace(); auto *inst = builder.CreateAlloca( - $_resultType->getPointerElementType(), $arraySize); + $_resultType->getPointerElementType(), addrSpace, $arraySize); }] # setAlignmentCode # [{ $res = inst; }];