https://github.com/halide/Halide
Raw File
Tip revision: b413bbac958b5eb8a0f8db6e998f132a6e7e4946 authored by Andrew Adams on 02 June 2024, 21:33:03 UTC
Merge remote-tracking branch 'origin/main' into abadams/constant_interval_simplifier
Tip revision: b413bba
ParallelRVar.h
#ifndef HALIDE_PARALLEL_RVAR_H
#define HALIDE_PARALLEL_RVAR_H

/** \file
 *
 * Method for checking if it's safe to parallelize an update
 * definition across a reduction variable.
 */

#include <string>

namespace Halide {
namespace Internal {

class Definition;

/** Returns whether or not Halide can prove that it is safe to
 * parallelize an update definition across a specific variable. If
 * this returns true, it's definitely safe. If this returns false, it
 * may still be safe, but Halide couldn't prove it.
 */
bool can_parallelize_rvar(const std::string &rvar,
                          const std::string &func,
                          const Definition &r);

}  // namespace Internal
}  // namespace Halide

#endif
back to top