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