https://github.com/halide/Halide
Raw File
Tip revision: 5f1e9d301c707a4eedf43c8d0bb1bd384e789bd5 authored by Z Stern on 16 February 2024, 06:37:10 UTC
Formatting.
Tip revision: 5f1e9d3
LoopPartitioningDirective.h
#ifndef HALIDE_LOOP_PARTITIONING_DIRECTIVE_H
#define HALIDE_LOOP_PARTITIONING_DIRECTIVE_H

/** \file
 * Defines the Partition enum.
 */

#include <string>

#include "Expr.h"
#include "Parameter.h"

namespace Halide {

/** Different ways to handle loops with a potentially optimizable boundary conditions. */
enum class Partition {
    /** Automatically let Halide decide on Loop Parititioning. */
    Auto,

    /** Disallow loop partitioning. */
    Never,

    /** Force partitioning of the loop, even in the tail cases of outer
     * partitioned loops. If Halide can't find a way to partition this loop, it
     * will raise an error. */
    Always
};

}  // namespace Halide

#endif  // HALIDE_LOOP_PARTITIONING_DIRECTIVE_H
back to top