1 //===- VectorToROCDL.h - Convert Vector to ROCDL dialect ---*- 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 #ifndef MLIR_CONVERSION_VECTORTOROCDL_VECTORTOROCDL_H_ 9 #define MLIR_CONVERSION_VECTORTOROCDL_VECTORTOROCDL_H_ 10 11 #include <memory> 12 13 namespace mlir { 14 class LLVMTypeConverter; 15 class OwningRewritePatternList; 16 class ModuleOp; 17 template <typename OpT> 18 class OperationPass; 19 20 /// Collect a set of patterns to convert from the GPU dialect to ROCDL. 21 void populateVectorToROCDLConversionPatterns( 22 LLVMTypeConverter &converter, OwningRewritePatternList &patterns); 23 24 /// Create a pass to convert vector operations to the ROCDL dialect. 25 std::unique_ptr<OperationPass<ModuleOp>> createConvertVectorToROCDLPass(); 26 27 } // namespace mlir 28 #endif // MLIR_CONVERSION_VECTORTOROCDL_VECTORTOROCDL_H_ 29