diff options
| author | Nicolas Vasilache <ntv@google.com> | 2019-08-18 18:54:50 -0700 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-08-18 18:55:14 -0700 |
| commit | c9f37fca379035b6334b50380ef05b00026de0cc (patch) | |
| tree | f5fc4eec3095f664286b0ac69b13a4ebe8e1ad26 /mlir/lib/LLVMIR/IR/LLVMDialect.cpp | |
| parent | 33a8642f537c409575a53b82f2607d730bdff639 (diff) | |
| download | bcm5719-llvm-c9f37fca379035b6334b50380ef05b00026de0cc.tar.gz bcm5719-llvm-c9f37fca379035b6334b50380ef05b00026de0cc.zip | |
Add alignment support for llvm.alloca
Extend the LLVM dialect AllocaOp with an alignment attribute.
PiperOrigin-RevId: 264068306
Diffstat (limited to 'mlir/lib/LLVMIR/IR/LLVMDialect.cpp')
| -rw-r--r-- | mlir/lib/LLVMIR/IR/LLVMDialect.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/mlir/lib/LLVMIR/IR/LLVMDialect.cpp b/mlir/lib/LLVMIR/IR/LLVMDialect.cpp index 199d40150dc..d051cc94d5b 100644 --- a/mlir/lib/LLVMIR/IR/LLVMDialect.cpp +++ b/mlir/lib/LLVMIR/IR/LLVMDialect.cpp @@ -131,7 +131,10 @@ static void printAllocaOp(OpAsmPrinter *p, AllocaOp &op) { op.getContext()); *p << op.getOperationName() << ' ' << *op.arraySize() << " x " << elemTy; - p->printOptionalAttrDict(op.getAttrs()); + if (op.alignment().hasValue() && op.alignment()->getSExtValue() != 0) + p->printOptionalAttrDict(op.getAttrs()); + else + p->printOptionalAttrDict(op.getAttrs(), {"alignment"}); *p << " : " << funcTy; } |
