Skip to content
Snippets Groups Projects

Draft: Copytape operator implementation

Open Olivier BICHLER requested to merge copytape into dev
Files
7
@@ -81,6 +81,27 @@ protected:
@@ -81,6 +81,27 @@ protected:
std::vector<Elts_t> mNbConsumedData;
std::vector<Elts_t> mNbConsumedData;
std::vector<Elts_t> mNbProducedData;
std::vector<Elts_t> mNbProducedData;
};
};
 
 
class OperatorCopyTapeImpl : public OperatorImpl {
 
public:
 
struct Chunk {
 
size_t src;
 
size_t size;
 
size_t dst;
 
};
 
 
OperatorCopyTapeImpl(const Operator& op, const std::string& backend = ""):
 
OperatorImpl(op, backend) {}
 
void copy(size_t src, size_t size, size_t dst);
 
virtual void forward() override;
 
void backward() override;
 
void setChunks(const std::vector<Chunk>& chunks) { mChunks = chunks; }
 
const std::vector<Chunk>& getChunks() const { return mChunks; }
 
virtual ~OperatorCopyTapeImpl() = default;
 
 
protected:
 
std::vector<Chunk> mChunks;
 
};
} // namespace Aidge
} // namespace Aidge
#endif /* AIDGE_BACKEND_OPERATORIMPL_H_ */
#endif /* AIDGE_BACKEND_OPERATORIMPL_H_ */
Loading