1 | #ifndef TRITON_INCLUDE_IR_CODEGEN_PIPELINE_H |
2 | #define TRITON_INCLUDE_IR_CODEGEN_PIPELINE_H |
3 | |
4 | // forward declaration |
5 | namespace triton { |
6 | namespace ir { |
7 | class module; |
8 | } |
9 | } // namespace triton |
10 | |
11 | namespace triton { |
12 | namespace codegen { |
13 | namespace transform { |
14 | |
15 | class pipeline { |
16 | public: |
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 | |
21 | private: |
22 | bool has_copy_async_; |
23 | int num_stages_; |
24 | }; |
25 | |
26 | } // namespace transform |
27 | } // namespace codegen |
28 | } // namespace triton |
29 | |
30 | #endif |
31 | |