Skip to content
Snippets Groups Projects
Commit d40ca864 authored by Jerome Hue's avatar Jerome Hue
Browse files

Save changes

parent 4a768192
No related branches found
No related tags found
No related merge requests found
Pipeline #57306 failed
......@@ -67,7 +67,7 @@ public:
namespace
{
static Registrar<Aidge::Heaviside_Op> registrarHeavisideImpl_cpu("cpu", Aidge::HeavisideImpl_cpu::create);
static Registrar<Heaviside_Op> registrarHeavisideImpl_cpu("cpu", Aidge::HeavisideImpl_cpu::create);
}
} // namespace Aidge
......
......@@ -75,16 +75,15 @@ void Aidge::AddImpl_cpu::forward() {
const auto& input = opTensor.getInput(i)->refCastFrom(inputsFallback[i], *opTensor.getOutput(0));
opInputs.push_back(input.getImpl()->rawPtr());
}
Log::info("Add Tensor input 0 : {}", opTensor.getInput(0)->toString());
Log::info("Add Tensor input 1 : {}", opTensor.getInput(1)->toString());
//Log::info("Add Tensor input 0 : {}", opTensor.getInput(0)->toString());
//Log::info("Add Tensor input 1 : {}", opTensor.getInput(1)->toString());
kernelFunc(opInputs,
inputsDims,
opTensor.getOutput(0)->size(),
opTensor.getOutput(0)->dims(),
getCPUPtr(opTensor.getRawOutput(0)));
Log::info("Add Tensor output : {}", opTensor.getOutput(0)->toString());
//Log::info("Add Tensor output : {}", opTensor.getOutput(0)->toString());
}
void Aidge::AddImpl_cpu::backward() {
......
......@@ -61,8 +61,6 @@ void Aidge::FCImpl_cpu::forward()
// Call kernel
Log::info("Call to FC kernel, with input data");
Log::info("{}\n", op_.getInput(0)->toString());
Log::info("and weights : ");
Log::info("{}\n\n", op_.getInput(1)->toString());
const auto batchSize = (input0.dims().size() > 1) ? input0.dims()[0] : 1;
kernelFunc(batchSize,
......
......@@ -52,7 +52,6 @@ void HeavisideImpl_cpu::forward() {
const auto inputDims0 = getBroadcastedDims(std::static_pointer_cast<Tensor>(mOp.getRawOutput(0))->dims(),
std::static_pointer_cast<Tensor>(mOp.getRawInput(0))->dims());
Log::info("opTensor input {}", opTensor.getInput(0)->toString());
// Call kernel
kernelFunc(inputDims0,
std::static_pointer_cast<Tensor>(mOp.getRawOutput(0))->dims(),
......
......@@ -48,8 +48,6 @@ void MulImpl_cpu::forward()
const std::vector<std::size_t> inputDims1 = getBroadcastedDims(std::static_pointer_cast<Tensor>(mOp.getRawOutput(0))->dims(),
std::static_pointer_cast<Tensor>(mOp.getRawInput(1))->dims());
Log::info("Mul node input 0 : {}", opTensor.getInput(0)->toString() );
Log::info("Mul node input 1 : {}", opTensor.getInput(1)->toString() );
// Call kernel
kernelFunc(inputDims0,
inputDims1,
......@@ -57,8 +55,6 @@ void MulImpl_cpu::forward()
getCPUPtr(mOp.getRawInput(0)),
getCPUPtr(mOp.getRawInput(1)),
getCPUPtr(mOp.getRawOutput(0)));
Log::info("Mul node output 0 : {}", opTensor.getOutput(0)->toString() );
}
void MulImpl_cpu::backward()
......
......@@ -41,9 +41,6 @@ void Aidge::SubImpl_cpu::forward() {
std::static_pointer_cast<Tensor>(mOp.getRawInput(0))->dims());
const std::vector<std::size_t> inputDims1 = getBroadcastedDims(std::static_pointer_cast<Tensor>(mOp.getRawOutput(0))->dims(),
std::static_pointer_cast<Tensor>(mOp.getRawInput(1))->dims());
Log::info("Sub node input 0 : {}", opTensor.getInput(0)->toString() );
Log::info("Sub node input 1 : {}", opTensor.getInput(1)->toString() );
// Call kernel
kernelFunc(inputDims0,
inputDims1,
......@@ -51,7 +48,4 @@ void Aidge::SubImpl_cpu::forward() {
getCPUPtr(mOp.getRawInput(0)),
getCPUPtr(mOp.getRawInput(1)),
getCPUPtr(mOp.getRawOutput(0)));
Log::info("Sub node output : {}", opTensor.getOutput(0)->toString() );
}
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