Skip to content
Snippets Groups Projects

Improved scheduling

Merged Olivier BICHLER requested to merge scheduling into dev
50 files
+ 1034
333
Compare changes
  • Side-by-side
  • Inline
Files
50
@@ -12,16 +12,17 @@
@@ -12,16 +12,17 @@
#ifndef AIDGE_CPU_OPERATOR_ADDIMPL_H_
#ifndef AIDGE_CPU_OPERATOR_ADDIMPL_H_
#define AIDGE_CPU_OPERATOR_ADDIMPL_H_
#define AIDGE_CPU_OPERATOR_ADDIMPL_H_
 
#include <cstddef> // std::size_t
 
#include <memory> // std::unique_ptr, std::make_unique
 
#include <string>
 
#include <vector>
 
#include "aidge/backend/OperatorImpl.hpp"
#include "aidge/backend/OperatorImpl.hpp"
#include "aidge/operator/Add.hpp"
#include "aidge/operator/Add.hpp"
#include "aidge/utils/Registrar.hpp"
#include "aidge/utils/Registrar.hpp"
#include "aidge/utils/Types.h"
#include "aidge/utils/Types.h"
#include "aidge/backend/cpu/data/GetCPUPtr.h"
#include <memory>
#include <vector>
namespace Aidge {
namespace Aidge {
// class Add_Op<2>;
// compute kernel registry for forward and backward
// compute kernel registry for forward and backward
class AddImplForward_cpu
class AddImplForward_cpu
@@ -33,7 +34,7 @@ class AddImplBackward_cpu
@@ -33,7 +34,7 @@ class AddImplBackward_cpu
class AddImpl_cpu : public OperatorImpl {
class AddImpl_cpu : public OperatorImpl {
public:
public:
AddImpl_cpu(const Add_Op& op) : OperatorImpl(op) {}
AddImpl_cpu(const Add_Op& op) : OperatorImpl(op, "cpu") {}
static std::unique_ptr<AddImpl_cpu> create(const Add_Op& op) {
static std::unique_ptr<AddImpl_cpu> create(const Add_Op& op) {
return std::make_unique<AddImpl_cpu>(op);
return std::make_unique<AddImpl_cpu>(op);
Loading