Skip to content
Snippets Groups Projects
Commit ffd6e77a authored by Maxence Naud's avatar Maxence Naud
Browse files

Merge branch 'broadcasting' of gitlab.eclipse.org:hrouis/aidge_backend_cpu into broadcasting

parents 83c6f795 90c2a6a1
No related branches found
No related tags found
2 merge requests!50version 0.2.0,!30add broadcasting for Arithmetic operators
Pipeline #39366 canceled
...@@ -68,7 +68,8 @@ void Aidge::AddImpl_cpu::forward() { ...@@ -68,7 +68,8 @@ void Aidge::AddImpl_cpu::forward() {
inputDims[idx] = dims[j]; inputDims[idx] = dims[j];
} }
inputsDims.push_back(inputDims); inputsDims.push_back(inputDims);
opInputs.push_back(getCPUPtr(mOp.getRawInput(i))); const auto& input = std::static_pointer_cast<Tensor>(mOp.getRawInput(i))->refCastFrom(inputsFallback[i], *std::static_pointer_cast<Tensor>(mOp.getRawOutput(0)));
opInputs.push_back(input.getImpl()->rawPtr());
} }
kernelFunc(opInputs, kernelFunc(opInputs,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment