mlir/include/mlir-c/Dialect/NVGPU.h

   1 //===-- mlir-c/Dialect/NVGPU.h - C API for NVGPU dialect --*- C -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM
   4 // Exceptions.
   5 // See https://llvm.org/LICENSE.txt for license information.
   6 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #ifndef MLIR_C_DIALECT_NVGPU_H
  11 #define MLIR_C_DIALECT_NVGPU_H
  12
  13 #include "mlir-c/IR.h"
  14 #include "mlir-c/Support.h"
  15
  16 #ifdef __cplusplus
  17 extern "C" {
  18 #endif
  19
  20 MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(NVGPU, nvgpu);
  21
  22 //===---------------------------------------------------------------------===//
  23 // TensorMapDescriptorType
  24 //===---------------------------------------------------------------------===//
  25
  26 MLIR_CAPI_EXPORTED bool mlirTypeIsANVGPUTensorMapDescriptorType(MlirType type);
  27
  28 MLIR_CAPI_EXPORTED MlirType mlirNVGPUTensorMapDescriptorTypeGet(
  29     MlirContext ctx, MlirType tensorMemrefType, int swizzle, int l2promo,
  30     int oobFill, int interleave);
  31
  32 #ifdef __cplusplus
  33 }
  34 #endif
  35
  36 #endif // MLIR_C_DIALECT_NVGPU_H