1#ifndef TRITON_INCLUDE_IR_CODEGEN_PIPELINE_H
2#define TRITON_INCLUDE_IR_CODEGEN_PIPELINE_H
3
4// forward declaration
5namespace triton {
6namespace ir {
7class module;
8}
9} // namespace triton
10
11namespace triton {
12namespace codegen {
13namespace transform {
14
15class pipeline {
16public:
17 pipeline(bool has_copy_async, int num_stages)
18 : has_copy_async_(has_copy_async), num_stages_(num_stages) {}
19 void run(ir::module &module);
20
21private:
22 bool has_copy_async_;
23 int num_stages_;
24};
25
26} // namespace transform
27} // namespace codegen
28} // namespace triton
29
30#endif
31