unroll.h

Go to the documentation of this file.

#pragma once

#include "tc/core/polyhedral/schedule_tree.h"

namespace tc {
namespace polyhedral {

/*
 * For each descendant band node of "st" in "root" (possibly including
 * "st" itself) and for each member of the band, mark it for unrolling
 * if the total number of instances executed by the band member
 * is smaller than or equal to the unroll value (if any).
 * Only unroll values greater than 1 can have any effect.
 *
 * The number of executed instances is estimated based the number
 * of values attained by the corresponding affine function given
 * fixed values of the outer bands and members.
 * The prefix schedule is therefore computed first,
 * taking into account the actual set of statement instances and
 * the filters along the path from "root" to "st".
 */
void markUnroll(
    detail::ScheduleTree* root,
    detail::ScheduleTree* st,
    uint64_t unroll);

} // namespace polyhedral
} // namespace tc