latest/doxygen/VarStructure_8hpp_source.html

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

#ifndef poputil_VarStructure_hpp

#define poputil_VarStructure_hpp


#include <functional>

#include <map>

#include <unordered_set>

#include <vector>


#include <poplar/Graph.hpp>


namespace poputil {


unsigned detectInnermostGrouping(const poplar::Graph &graph,

                                 const poplar::Tensor &t);


using GroupingInfo = std::pair<unsigned, unsigned>;


std::vector<GroupingInfo> detectDimGroupings(const poplar::Graph &graph,

                                             const poplar::Tensor &t);


poplar::Tensor

createPartitionableTensor(poplar::Graph &graph, const poplar::Type &type,

                          const std::vector<std::size_t> &shape,

                          const std::vector<std::size_t> &nPartitions,

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


void iterateTensorPartitions(

    const poplar::Tensor &t, const std::vector<std::size_t> &nPartitions,

    const std::function<void(const std::vector<std::size_t> &i,

                             const poplar::Tensor &s)> &f);


} // end namespace poputil


#endif // poputil_VarStructure_hpp

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::Type
Class representing device data types.
Definition: Type.hpp:42

poputil
General utility functions for building graphs.
Definition: GfloatExprUtil.hpp:23

poputil::GroupingInfo
std::pair< unsigned, unsigned > GroupingInfo
Grouped dimension info.
Definition: VarStructure.hpp:35

poputil::detectDimGroupings
std::vector< GroupingInfo > detectDimGroupings(const poplar::Graph &graph, const poplar::Tensor &t)
Find all grouped dimensions from the innermost grouped dimension moving outwards, returning groupings...

poputil::detectInnermostGrouping
unsigned detectInnermostGrouping(const poplar::Graph &graph, const poplar::Tensor &t)
Detect if the tensor t has a grouping in its innermost dimension.

poputil::iterateTensorPartitions
void iterateTensorPartitions(const poplar::Tensor &t, const std::vector< std::size_t > &nPartitions, const std::function< void(const std::vector< std::size_t > &i, const poplar::Tensor &s)> &f)
Iterate a function over the partitions of a tensor.

poputil::createPartitionableTensor
poplar::Tensor createPartitionableTensor(poplar::Graph &graph, const poplar::Type &type, const std::vector< std::size_t > &shape, const std::vector< std::size_t > &nPartitions, const poplar::DebugContext &debugContext={})
Create a tensor with the given shape, so that when it is partitioned into slices according to the giv...