MLIR  16.0.0git
Passes.h
Go to the documentation of this file.
1 //===- Passes.h - Async pass entry points -----------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This header file defines prototypes that expose pass constructors.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef MLIR_DIALECT_ASYNC_PASSES_H_
14 #define MLIR_DIALECT_ASYNC_PASSES_H_
15 
16 #include "mlir/Pass/Pass.h"
17 
18 namespace mlir {
19 class ModuleOp;
20 
21 std::unique_ptr<Pass> createAsyncParallelForPass();
22 
23 std::unique_ptr<Pass> createAsyncParallelForPass(bool asyncDispatch,
24  int32_t numWorkerThreads,
25  int32_t minTaskSize);
26 
27 std::unique_ptr<OperationPass<ModuleOp>> createAsyncToAsyncRuntimePass();
28 
29 std::unique_ptr<Pass> createAsyncRuntimeRefCountingPass();
30 
31 std::unique_ptr<Pass> createAsyncRuntimeRefCountingOptPass();
32 
33 std::unique_ptr<Pass> createAsyncRuntimePolicyBasedRefCountingPass();
34 
35 //===----------------------------------------------------------------------===//
36 // Registration
37 //===----------------------------------------------------------------------===//
38 
39 /// Generate the code for registering passes.
40 #define GEN_PASS_REGISTRATION
41 #include "mlir/Dialect/Async/Passes.h.inc"
42 
43 } // namespace mlir
44 
45 #endif // MLIR_DIALECT_ASYNC_PASSES_H_
Include the generated interface declarations.
std::unique_ptr< Pass > createAsyncRuntimeRefCountingPass()
std::unique_ptr< OperationPass< ModuleOp > > createAsyncToAsyncRuntimePass()
std::unique_ptr< Pass > createAsyncRuntimePolicyBasedRefCountingPass()
std::unique_ptr< Pass > createAsyncParallelForPass()
std::unique_ptr< Pass > createAsyncRuntimeRefCountingOptPass()