//===- TensorToSPIRVPass.h - Tensor to SPIR-V Passes --------*- C++ -*-===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// // // Provides passes to convert Tensor dialect to SPIR-V dialect. // //===----------------------------------------------------------------------===// #ifndef MLIR_CONVERSION_TENSORTOSPIRV_TENSORTOSPIRVPASS_H #define MLIR_CONVERSION_TENSORTOSPIRV_TENSORTOSPIRVPASS_H #include "mlir/Pass/Pass.h" namespace mlir { class ModuleOp; #define GEN_PASS_DECL_CONVERTTENSORTOSPIRV #include "mlir/Conversion/Passes.h.inc" /// Creates a pass to convert Tensor ops to SPIR-V ops. std::unique_ptr<OperationPass<>> createConvertTensorToSPIRVPass(); } // namespace mlir #endif // MLIR_CONVERSION_TENSORTOSPIRV_TENSORTOSPIRVPASS_H