Fix build break on debug build (#1427)

Fix debug build break for pure acl runtime

Signed-off-by: Chunseok Lee <chunseok.lee@samsung.com>

diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc
index 0d07d2c..0b580d9 100644 (file)
--- a/runtimes/pure_arm_compute/src/compilation.cc
+++ b/runtimes/pure_arm_compute/src/compilation.cc
@@ -313,7 +313,8 @@ void Planner::visit(const ::internal::tflite::op::Add::Node &node)
 
   param.activation = static_cast<FuseCode>(_ctx.at(activation_index).asScala<int32_t>());
 
-  auto stage = [param](const IAllocationContext &ctx, IExecutionBuilder &builder) {
+  auto stage = [param, lhs_shape, rhs_shape](const IAllocationContext &ctx,
+                                             IExecutionBuilder &builder) {
     auto ofm_alloc = ctx.at(::internal::tflite::operand::Index{param.ofm_index});
     auto lhs_alloc = ctx.at(::internal::tflite::operand::Index{param.lhs_index});
     auto rhs_alloc = ctx.at(::internal::tflite::operand::Index{param.rhs_index});
@@ -334,9 +335,9 @@ void Planner::visit(const ::internal::tflite::op::Add::Node &node)
     else
     {
       // NOTE SimpleAritemeticAdditionLayer does not support broadcasting
-      assert(lhs_shape.depth() == rhs_shape.depth());
-      assert(lhs_shape.height() == rhs_shape.height());
-      assert(lhs_shape.width() == rhs_shape.width());
+      assert(lhs_shape.C == rhs_shape.C);
+      assert(lhs_shape.H == rhs_shape.H);
+      assert(lhs_shape.W == rhs_shape.W);
 
       auto l = make_layer<SimpleArithmeticAdditionLayer>();