ConvPreplan

#include <poplin/ConvPreplan.hpp>

Functions and data types to support performing convolution preplanning

namespace poplin

Linear algebra functions.

Decomposition of a matrix into an lower triangular matrix L and upper triangular matrix U.

Functions

void preplan(const std::set<ConvPlanParams> &convs, const std::set<MatMulPlanParams> &matmuls, PlanningCache &cache)

Plan the specified convolutions & matmuls.

All entries must have matching machine parameters.

All entries must have matching machine parameters.

Parameters

  • convs – A set of tuples of:

    • conv-specific target for tile / IPU sizing

    • convolution parameters

    • implementation options. See createWeights().

  • matmuls – A set of tuples of:

    • matmul-specific target for tile / IPU sizing

    • convolution parameters

    • implementation options. See createWeights().

  • cache – The planning cache to update.