From dba5e80d76ff31c6b4afc18c5009386d1c9fbab4 Mon Sep 17 00:00:00 2001
From: Jerome Hue <jerome.hue@cea.fr>
Date: Wed, 19 Feb 2025 14:53:10 +0100
Subject: [PATCH] Fix StackOp forward()
Call OperatorTensor::forward() instead of Operator::forward(), ensuring dims are
forwarded.
---
src/operator/Stack.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/operator/Stack.cpp b/src/operator/Stack.cpp
index a938f470d..17502e0f4 100644
--- a/src/operator/Stack.cpp
+++ b/src/operator/Stack.cpp
@@ -132,7 +132,7 @@ std::set<std::string> StackOp::getAvailableBackends() const {
}
void StackOp::forward() {
- Operator::forward();
+ OperatorTensor::forward();
++forwardStep();
}
--
GitLab