• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1diff --git a/src/cpu/operators/CpuGemmDirectConv2d.cpp b/src/cpu/operators/CpuGemmDirectConv2d.cpp
2index 75c057e45..4d4d0e272 100644
3--- a/src/cpu/operators/CpuGemmDirectConv2d.cpp
4+++ b/src/cpu/operators/CpuGemmDirectConv2d.cpp
5@@ -193,6 +193,8 @@ void CpuGemmDirectConv2d::run(ITensorPack &tensors)
6     _gemm_asm_func->run(tensors);
7     if(_run_activation)
8     {
9-        _activation_func->run(tensors);
10+        auto io = tensors.get_tensor(ACL_DST);
11+        ITensorPack pack{ { ACL_SRC, io }, { ACL_DST, io } };
12+        _activation_func->run(pack);
13     }
14 }
15