| // Copyright 2024 The IREE Authors |
| // |
| // Licensed 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 |
| |
| #ifndef IREE_COMPILER_PLUGINS_TARGET_ROCM_ROCMTARGETFEATURES_H_ |
| #define IREE_COMPILER_PLUGINS_TARGET_ROCM_ROCMTARGETFEATURES_H_ |
| |
| #include "iree/compiler/Dialect/HAL/Target/TargetBackend.h" |
| #include "llvm/IR/Module.h" |
| #include "llvm/Target/TargetMachine.h" |
| |
| namespace mlir::iree_compiler::IREE::HAL { |
| |
| // Returns the list of supported mma types (mfma/wmma). |
| ArrayAttr getROCMSupportedMmaAttrs(MLIRContext *context, StringRef targetArch); |
| |
| } // namespace mlir::iree_compiler::IREE::HAL |
| |
| #endif // IREE_COMPILER_PLUGINS_TARGET_ROCM_ROCMTARGETFEATURES_H_ |
