MLIR  20.0.0git
NVGPU.h
Go to the documentation of this file.
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 
21 
22 //===---------------------------------------------------------------------===//
23 // TensorMapDescriptorType
24 //===---------------------------------------------------------------------===//
25 
27 
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
MLIR_CAPI_EXPORTED bool mlirTypeIsANVGPUTensorMapDescriptorType(MlirType type)
Definition: NVGPU.cpp:19
MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(NVGPU, nvgpu)
MLIR_CAPI_EXPORTED MlirType mlirNVGPUTensorMapDescriptorTypeGet(MlirContext ctx, MlirType tensorMemrefType, int swizzle, int l2promo, int oobFill, int interleave)
Definition: NVGPU.cpp:23
#define MLIR_CAPI_EXPORTED
Definition: Support.h:46