WIP

NVIDIA · Dec 24, 2024 · f5f5c53 · f5f5c53
1 parent 9fc0595
commit f5f5c53
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/csrc/preseg_passes/remove_bcast_squeeze.cpp b/csrc/preseg_passes/remove_bcast_squeeze.cpp
@@ -163,13 +163,13 @@ TensorView* replayAxisOp(
   switch (simple_op_type) {
     case AxisOp::PRESERVE:
       // This is equivalent to a set Op
-      replacement = tv;
+      return tv;
       break;
     case AxisOp::SQUEEZE:
-      replacement = squeeze(tv, nonPreservedDims(axis_ops));
+      return squeeze(tv, nonPreservedDims(axis_ops));
       break;
     case AxisOp::BROADCAST:
-      replacement = broadcast(tv, nonPreservedDims(axis_ops));
+      return broadcast(tv, nonPreservedDims(axis_ops));
       break;
   }
 }