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

Merge branch 'dev' into 'sub-backward'

# Conflicts:
#   unit_tests/operator/Test_MetaOperator.cpp
parents ff512139 4cea3405
No related branches found
No related tags found
1 merge request!138Implement backward function for Sub
Pipeline #67076 passed
......@@ -33,8 +33,6 @@ using AddImpl_cpu = OperatorImpl_cpu<Add_Op,
const std::vector<std::size_t>&,
const std::vector<std::size_t>&,
const void*,
const void*,
const void*,
void*,
void*)
>;
......
......@@ -154,15 +154,11 @@ void AddImpl_cpu_backward_kernel(const std::size_t input0Length,
const std::vector<std::size_t>& dims0,
const std::vector<std::size_t>& dims1,
const std::vector<std::size_t>& outputDims,
const void* input0_,
const void* input1_,
const void* grad_output_,
void* gradientInput0_,
void* gradientInput1_)
{
// TODO: Remove input0/1 from the function
const I* input0 = static_cast<const I*>(input0_);
const I* input1 = static_cast<const I*>(input1_);
const O* gradOutput = static_cast<const O*>(grad_output_);
auto* gradInput0 = static_cast<I*>(gradientInput0_);
auto* gradInput1 = static_cast<I*>(gradientInput1_);
......
......@@ -73,8 +73,6 @@ void Aidge::AddImpl_cpu::backward() {
in0->dims(),
in1->dims(),
out0grad->dims(),
getCPUPtr(in0),
getCPUPtr(in1),
getCPUPtr(out0grad),
getCPUPtr(in0grad),
getCPUPtr(in1grad));
......
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