aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CL/functions/CLScatter.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/CL/functions/CLScatter.cpp')
-rw-r--r--src/runtime/CL/functions/CLScatter.cpp11
1 files changed, 9 insertions, 2 deletions
diff --git a/src/runtime/CL/functions/CLScatter.cpp b/src/runtime/CL/functions/CLScatter.cpp
index e1de92968a..e16fcc4ccc 100644
--- a/src/runtime/CL/functions/CLScatter.cpp
+++ b/src/runtime/CL/functions/CLScatter.cpp
@@ -62,10 +62,17 @@ void CLScatter::configure(const CLCompileContext &compile_context,
ICLTensor *output,
const ScatterInfo &info)
{
- ARM_COMPUTE_ERROR_ON_NULLPTR(src, indices, output);
+ ARM_COMPUTE_ERROR_ON_NULLPTR(updates, indices, output);
_impl->op = std::make_unique<OperatorType>();
- _impl->op->configure(compile_context, src->info(), updates->info(), indices->info(), output->info(), info);
+ if (src)
+ { // Src not nullptr.
+ _impl->op->configure(compile_context, src->info(), updates->info(), indices->info(), output->info(), info);
+ }
+ else
+ {
+ _impl->op->configure(compile_context, nullptr, updates->info(), indices->info(), output->info(), info);
+ }
_impl->run_pack = {{ACL_SRC_0, src}, {ACL_SRC_1, updates}, {ACL_SRC_2, indices}, {ACL_DST, output}};
}