Fix typo

xmos · Aug 2, 2024 · 4a8bf20 · 4a8bf20
1 parent c9e7748
commit 4a8bf20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xformer/Transforms/ReplaceConcat.cpp b/xformer/Transforms/ReplaceConcat.cpp
@@ -62,8 +62,8 @@ struct SplitConcatPattern : public OpRewritePattern<TFL::ConcatenationOp> {
     auto outputType = concatOp.getOutput().getType().cast<RankedTensorType>();
     Type elementType = outputType.getElementType();
     ArrayRef<int64_t> outputShape = outputType.getShape();
-    int axis = concatOp.getAxis();
-    axis = -1 ? outputType.getRank() - 1 : axis;
+    const int axis = concatOp.getAxis() == -1 ? outputType.getRank() - 1
+                                              : concatOp.getAxis();
 
     int axisShape = 0;
     for (int i = 0; i < CONCAT_OP_MAX_INPUTS; i++) {