Function tvm::sym::IterMapSimplify

Function tvm::sym::IterMapSimplify#

Function Documentation#

ffi::Array<PrimExpr> tvm::sym::IterMapSimplify(const ffi::Array<PrimExpr> &indices, const ffi::Map<PrimVar, Range> &input_iters, const PrimExpr &input_pred, IterMapLevel check_level, const sym::Analyzer &analyzer, bool simplify_trivial_iterators = true)#

Use IterVarMap detector to rewrite and simplify the indices.

Parameters:

  • indices – The indices to detect pattern for.

  • input_iters – Map from variable to iterator’s range.

  • input_pred – The predicate constraints on the input iterators

  • check_level – The iter mapping checking level.

  • analyzer – Analyzer used to get context information.

  • simplify_trivial_iterators – If true, iterators with unit extents are simplified

Returns:

The indices after rewrite