From f5f5c5361f6e02ea6af4b4cfc05b2c0fff40d4be Mon Sep 17 00:00:00 2001 From: jjsjann123 Date: Tue, 24 Dec 2024 12:54:07 -0800 Subject: [PATCH] WIP --- csrc/preseg_passes/remove_bcast_squeeze.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/csrc/preseg_passes/remove_bcast_squeeze.cpp b/csrc/preseg_passes/remove_bcast_squeeze.cpp index ce0706dd74c..0dfd782686d 100644 --- 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; } }