https://github.com/halide/Halide
Raw File
Tip revision: e848ee8b800a06d9a1d7135482c6610d18b468ec authored by Andrew Adams on 21 February 2024, 18:54:24 UTC
Merge remote-tracking branch 'origin/main' into abadams/cache_tighten_producer_consumer_nodes
Tip revision: e848ee8
ExtractTileOperations.h
#ifndef HALIDE_EXTRACT_TILE_OPERATIONS_H
#define HALIDE_EXTRACT_TILE_OPERATIONS_H

/** \file
 * Defines the lowering pass that injects calls to tile intrinsics that support
 * AMX instructions.
 */

#include "Expr.h"

namespace Halide {
namespace Internal {

/** Rewrite any AMX tile operations that have been stored in the AMXTile memory
 * type as intrinsic calls, to be used in the X86 backend. */
Stmt extract_tile_operations(const Stmt &s);

}  // namespace Internal
}  // namespace Halide

#endif
back to top