compiler/src/iree/compiler/Dialect/HAL/Conversion/TypeConverter.h

// Copyright 2019 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_DIALECT_HAL_CONVERSION_TYPECONVERTER_H_
#define IREE_COMPILER_DIALECT_HAL_CONVERSION_TYPECONVERTER_H_

#include <vector>

#include "iree/compiler/Dialect/HAL/Conversion/ConversionDialectInterface.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/Transforms/DialectConversion.h"

namespace mlir::iree_compiler {

class HALTypeConverter : public TypeConverter {
public:
  explicit HALTypeConverter(
      ArrayRef<const HALConversionDialectInterface *> conversionInterfaces);

  // TODO(benvanik): signature conversion for output buffers.

  // Returns true if the given |type| maps to !hal.buffer_view by default.
  // hal.tensor.import/export can be used by frontends to map to other types.
  static bool shouldConvertToBufferView(Type type) {
    if (auto tensorType = dyn_cast<TensorType>(type)) {
      return tensorType.getElementType().isIntOrFloat();
    }
    return false;
  }

private:
  // The set of dialect conversion interfaces we should query to convert types.
  std::vector<const HALConversionDialectInterface *> conversionInterfaces;
};

} // namespace mlir::iree_compiler

#endif // IREE_COMPILER_DIALECT_HAL_CONVERSION_TYPECONVERTER_H_