xref: /llvm-project/mlir/include/mlir-c/Dialect/NVGPU.h (revision 4f88c2311130791cf69da34b743b1b3ba7584a7b)
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
37