Revision 3fbb12aa928a87a4c4e69a95eaf6f5ee02958600 authored by John Lawson on 09 February 2021, 18:00:02 UTC, committed by GitHub on 09 February 2021, 18:00:02 UTC
* Add support for AVX512 BF16 dot product * Match on f32*f32 * Remove f32 check
1 parent 3e034d6
SkipStages.h
#ifndef HALIDE_SKIP_STAGES
#define HALIDE_SKIP_STAGES
#include <string>
#include <vector>
#include "Expr.h"
/** \file
* Defines a pass that dynamically avoids realizing unnecessary stages.
*/
namespace Halide {
namespace Internal {
/** Avoid computing certain stages if we can infer a runtime condition
* to check that tells us they won't be used. Does this by analyzing
* all reads of each buffer allocated, and inferring some condition
* that tells us if the reads occur. If the condition is non-trivial,
* inject ifs that guard the production. */
Stmt skip_stages(Stmt s, const std::vector<std::string> &order);
} // namespace Internal
} // namespace Halide
#endif
Computing file changes ...
