Skip to content
Snippets Groups Projects
Commit 90fa4446 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fixed missing changes

parent 86ccc39f
No related branches found
No related tags found
2 merge requests!10Unified interface for attributes,!6Tensor setter getter
Pipeline #32456 failed
...@@ -18,16 +18,17 @@ ...@@ -18,16 +18,17 @@
#include "aidge/utils/Types.h" #include "aidge/utils/Types.h"
#include <memory> #include <memory>
#include <vector> #include <vector>
#include <array>
namespace Aidge { namespace Aidge {
// class Scaling_Op; // class Scaling_Op;
// compute kernel registry for forward and backward // compute kernel registry for forward and backward
class ScalingImplForward_cpu class ScalingImplForward_cpu
: public Registrable<ScalingImplForward_cpu, std::tuple<DataType, DataType>, void(const Scaling_Op::Attributes&, std::size_t, const void*, void*)> { : public Registrable<ScalingImplForward_cpu, std::tuple<DataType, DataType>, void(const Scaling_Op::Attrs&, std::size_t, const void*, void*)> {
}; };
class ScalingImplBackward_cpu class ScalingImplBackward_cpu
: public Registrable<ScalingImplBackward_cpu, std::tuple<DataType, DataType>, void(const Scaling_Op::Attributes&, std::size_t, const void*, void*)> { : public Registrable<ScalingImplBackward_cpu, std::tuple<DataType, DataType>, void(const Scaling_Op::Attrs&, std::size_t, const void*, void*)> {
}; };
class ScalingImpl_cpu : public OperatorImpl { class ScalingImpl_cpu : public OperatorImpl {
......
...@@ -18,14 +18,14 @@ ...@@ -18,14 +18,14 @@
namespace Aidge { namespace Aidge {
template <class I, class O> template <class I, class O>
void ScalingImpl_cpu_forward_kernel(const Scaling_Op::Attributes& attrs, void ScalingImpl_cpu_forward_kernel(const Scaling_Op::Attrs& attrs,
std::size_t inputLenght, std::size_t inputLenght,
const void* input_, const void* input_,
void* output_) { void* output_) {
const I* input = static_cast<const I*>(input_); const I* input = static_cast<const I*>(input_);
O* output = static_cast<O*>(output_); O* output = static_cast<O*>(output_);
I scalingFactor = static_cast<I>(std::get<0>(attrs)); const I& scalingFactor = static_cast<const I&>(std::get<0>(attrs));
for (std::size_t i = 0; i < inputLenght; ++i) { for (std::size_t i = 0; i < inputLenght; ++i) {
output[i] = input[i] * scalingFactor; output[i] = input[i] * scalingFactor;
......
...@@ -68,7 +68,7 @@ void Aidge::ScalingImpl_cpu::forward() { ...@@ -68,7 +68,7 @@ void Aidge::ScalingImpl_cpu::forward() {
mOp.getOutput(0)->dataType()}); mOp.getOutput(0)->dataType()});
// Call kernel // Call kernel
kernelFunc(mOp.getParams(), kernelFunc(mOp.getStaticAttributes(),
std::static_pointer_cast<Tensor>(mOp.getInput(0))->size(), std::static_pointer_cast<Tensor>(mOp.getInput(0))->size(),
mOp.getInput(0)->getImpl()->rawPtr(), mOp.getInput(0)->getImpl()->rawPtr(),
mOp.getOutput(0)->getImpl()->rawPtr()); mOp.getOutput(0)->getImpl()->rawPtr());
......
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