Poplar and PopLibs
Zero.hpp
Go to the documentation of this file.
1// Copyright (c) 2016 Graphcore Ltd. All rights reserved.
8#ifndef popops_Zero_hpp
9#define popops_Zero_hpp
10
11#include "poplar/Graph.hpp"
12#include "poplar/Program.hpp"
13#include <vector>
14
15namespace popops {
16
26void zero(poplar::Graph &graph, poplar::Tensor t,
27 const std::vector<poplar::Interval> &tileRegions, unsigned tile,
28 poplar::ComputeSet zeroCS);
29
38void zero(poplar::Graph &graph, const poplar::Tensor &t, unsigned tile,
39 poplar::ComputeSet zeroCS);
40
52void zero(poplar::Graph &graph, const poplar::Tensor &t,
53 const std::vector<std::vector<poplar::Interval>> &mapping,
54 poplar::ComputeSet zeroCS);
55
64void zero(poplar::Graph &graph, const poplar::Tensor &t,
65 poplar::program::Sequence &prog,
66 const poplar::DebugContext &debugContext = {});
67
68} // namespace popops
69
70#endif // popops_Zero_hpp
poplar::ComputeSet
A reference to a compute set within a graph.
Definition: GraphElements.hpp:131
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::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::zero
void zero(poplar::Graph &graph, poplar::Tensor t, const std::vector< poplar::Interval > &tileRegions, unsigned tile, poplar::ComputeSet zeroCS)
Append vertices to compute set zeroCS which zeroes elements in tileRegions of tensor t which reside o...