Skip to content

Commit f4664aa

Browse files
committed
[mlir] Use memref's alignment attribute directly.
1 parent 3b5aff5 commit f4664aa

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

mlir/lib/Conversion/MemRefToSPIRV/MemRefToSPIRV.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -511,8 +511,7 @@ calculateMemoryRequirements(Value accessedPtr, LoadOrStoreOp loadOrStoreOp) {
511511
Operation *memrefAccessOp = loadOrStoreOp.getOperation();
512512
auto memrefMemAccess = memrefAccessOp->getAttrOfType<spirv::MemoryAccessAttr>(
513513
spirv::attributeName<spirv::MemoryAccess>());
514-
auto memrefAlignment =
515-
memrefAccessOp->getAttrOfType<IntegerAttr>("alignment");
514+
auto memrefAlignment = loadOrStoreOp.getAlignmentAttr();
516515
if (memrefMemAccess && memrefAlignment)
517516
return MemoryRequirements{memrefMemAccess, memrefAlignment};
518517

0 commit comments

Comments
 (0)