WIP

NVIDIA · Dec 24, 2024 · 9fc0595 · 9fc0595
1 parent 26e9330
commit 9fc0595
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/preseg_passes/remove_bcast_squeeze.cpp b/csrc/preseg_passes/remove_bcast_squeeze.cpp
@@ -157,7 +157,7 @@ std::vector<bool> nonPreservedDims(const AxisOps& ops) {
 }
 
 TensorView* replayAxisOp(
-    AxisOps simple_op_type,
+    AxisOp simple_op_type,
     const AxisOps& axis_ops,
     TensorView* tv) {
   switch (simple_op_type) {