-
Notifications
You must be signed in to change notification settings - Fork 48
Expand file tree
/
Copy pathAIRMatmulCodegen.h
More file actions
33 lines (26 loc) · 1.01 KB
/
AIRMatmulCodegen.h
File metadata and controls
33 lines (26 loc) · 1.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
//===- AIRMatmulCodegen.h ---------------------------------------*- C++ -*-===//
//
// Copyright (C) 2026, Advanced Micro Devices, Inc. All rights reserved.
// SPDX-License-Identifier: MIT
//
//===----------------------------------------------------------------------===//
//
// AIRMatmulCodegen: single public matmul codegen pass. Orchestrates the
// internal phases (launch tile, pack, K-tile, core tile, prologue/epilogue,
// bufferization, vectorize) in fixed order. Internal phases are exposed as
// free functions in their respective headers.
//
//===----------------------------------------------------------------------===//
#ifndef AIR_MATMUL_CODEGEN_H
#define AIR_MATMUL_CODEGEN_H
#include "air/Transform/PassDetail.h"
#include "mlir/Pass/Pass.h"
#include <memory>
namespace xilinx {
namespace air {
std::unique_ptr<mlir::Pass> createAIRMatmulCodegenPass();
std::unique_ptr<mlir::Pass>
createAIRMatmulCodegenPass(const AIRMatmulCodegenOptions &);
} // namespace air
} // namespace xilinx
#endif // AIR_MATMUL_CODEGEN_H