Skip to content

Possible mistake in Pow Operator backward pass

What commit version of aidge do you use

  • aidge_backend_cpu: 0.2.3

Problem description

The backward() function of the Pow operator implementation is looking for a forward kernel function. That seems wrong.

See PowImpl.cpp :

void Aidge::PowImpl_cpu::backward() {
    // Find the correct kernel type
    const Pow_Op& op_ = dynamic_cast<const Pow_Op&>(mOp);
    auto kernelFunc = Registrar<PowImplForward_cpu>::create({
        op_.getOutput(0)->grad()->dataType(),
        op_.getInput(0)->grad()->dataType(),
        op_.getInput(1)->grad()->dataType()});