diff --git a/codegen/compiler/src/Quidditch/Dialect/Snitch/Transforms/LowerL1Allocations.cpp b/codegen/compiler/src/Quidditch/Dialect/Snitch/Transforms/LowerL1Allocations.cpp index 61f61f3f..1be6c642 100644 --- a/codegen/compiler/src/Quidditch/Dialect/Snitch/Transforms/LowerL1Allocations.cpp +++ b/codegen/compiler/src/Quidditch/Dialect/Snitch/Transforms/LowerL1Allocations.cpp @@ -55,11 +55,9 @@ void LowerL1Allocations::runOnOperation() { builder.setInsertionPoint(allocOp); MemRefType memRefType = allocOp.getType(); // Note: This assumes bitWidth == alignment == size. + // Since we use a scratchpad, align to size of element in bytes. uint64_t bitWidth = memRefType.getElementTypeBitWidth(); - if (std::optional alignment = allocOp.getAlignment()) - offset = llvm::alignTo(offset, *alignment); - else - offset = llvm::alignTo(offset, llvm::divideCeil(bitWidth, 8)); + offset = llvm::alignTo(offset, llvm::divideCeil(bitWidth, 8)); auto byteShift = builder.create(allocOp.getLoc(), offset);