Skip to content

Commit 89b2f1c

Browse files
committed
missed one line
1 parent b0ded74 commit 89b2f1c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

csrc/runtime/allocations.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -654,6 +654,7 @@ class BackwardTraverseFromAllocToLogical {
654654
// padded extent. We use expr_eval to slice out the padding session, so that
655655
// the logical domain would remain the correct extent instead of padding
656656
// sizes.
657+
int64_t in_extent = ee_.evaluate(in->extent()).as<int64_t>();
657658
if (areDimsToBeMergedContiguous(tensor_, new_shape)) {
658659
tensor_ = tensor_.view(new_shape);
659660
// slice for non-divisible split

0 commit comments

Comments
 (0)