My Project
GPUToROCDLPass.h
Go to the documentation of this file.
1 //===- GPUToROCDLPass.h - Convert GPU kernel to ROCDL dialect ---*- C++ -*-===//
2 //
3 // Part of the MLIR 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 #ifndef MLIR_CONVERSION_GPUTOROCDL_GPUTOROCDLPASS_H_
9 #define MLIR_CONVERSION_GPUTOROCDL_GPUTOROCDLPASS_H_
10 
11 #include <memory>
12 
13 namespace mlir {
14 
15 class ModuleOp;
16 template <typename OpT> class OpPassBase;
17 
19 std::unique_ptr<OpPassBase<ModuleOp>> createLowerGpuOpsToROCDLOpsPass();
20 
21 } // namespace mlir
22 
23 #endif // MLIR_CONVERSION_GPUTOROCDL_GPUTOROCDLPASS_H_
Definition: InferTypeOpInterface.cpp:20
std::unique_ptr< OpPassBase< ModuleOp > > createLowerGpuOpsToROCDLOpsPass()
Creates a pass that lowers GPU dialect operations to ROCDL counterparts.
Definition: LowerGpuOpsToROCDLOps.cpp:69