latest/doxygen/Loop_8hpp_source.html

// Copyright (c) 2020 Graphcore Ltd. All rights reserved.

#ifndef popops_Loop_hpp

#define popops_Loop_hpp


#include <poplar/Graph.hpp>

#include <poplar/Program.hpp>

#include <poputil/DebugInfo.hpp>

#include <poputil/VertexTemplates.hpp>


namespace popops {


using CountedLoopBodyType =

    std::function<poplar::program::Program(const poplar::Tensor &)>;


poplar::program::Sequence

countedLoop(poplar::Graph &graph, std::size_t begin, std::size_t end,

            size_t step, const CountedLoopBodyType &body,

            const poplar::DebugContext &debugContext = {});


poplar::program::Sequence

countedLoop(poplar::Graph &graph, std::size_t count,

            const CountedLoopBodyType &body,

            const poplar::DebugContext &debugContext = {});


poplar::Tensor addForLoopCounterVertex(poplar::Graph &graph,

                                       const poplar::Tensor &count,

                                       const poplar::Tensor &countLimit,

                                       int countStep, unsigned tile,

                                       poplar::program::Sequence &prog,

                                       const poplar::DebugContext &di);


poplar::program::Sequence

countedForLoop(poplar::Graph &graph, const poplar::Tensor &count,

               int initialCount, const poplar::Tensor &countLimit,

               int countStep, const poplar::program::Program &body,

               const poplar::DebugContext &debugContext = {});


poplar::program::Sequence

countedForLoop(poplar::Graph &graph, int initialCount,

               const poplar::Tensor &countLimit, int countStep,

               const poplar::program::Program &body,

               const poplar::DebugContext &debugContext = {});


} // namespace popops


#endif // popops_Loop_hpp

DebugInfo.hpp
Poplibs generic debug info structure.

VertexTemplates.hpp
Generate a string describing a vertex type.

poplar::DebugContext
DebugContext gathers the common external parameters of the context of an operation.
Definition: DebugContext.hpp:221

poplar::Graph
This class represents a graph program to be executed on the IPU.
Definition: Graph.hpp:52

poplar::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38

poplar::program::Program
This class represents a control program that executes operations on the graph.
Definition: Program.hpp:30

poplar::program::Sequence
Program that executes a sequence of programs.
Definition: Program.hpp:77

popops
Common functions, such as elementwise and reductions.
Definition: AllTrue.hpp:15

popops::countedLoop
poplar::program::Sequence countedLoop(poplar::Graph &graph, std::size_t begin, std::size_t end, size_t step, const CountedLoopBodyType &body, const poplar::DebugContext &debugContext={})
Create a loop program with constant initial count, increment and end value.

popops::countedForLoop
poplar::program::Sequence countedForLoop(poplar::Graph &graph, const poplar::Tensor &count, int initialCount, const poplar::Tensor &countLimit, int countStep, const poplar::program::Program &body, const poplar::DebugContext &debugContext={})
Create a for-loop program with constant initial count and increment, and a tensor as the end value.