37 lines · c
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 LLVM4// Exceptions.5// See https://llvm.org/LICENSE.txt for license information.6// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception7//8//===----------------------------------------------------------------------===//9 10#ifndef MLIR_C_DIALECT_NVGPU_H11#define MLIR_C_DIALECT_NVGPU_H12 13#include "mlir-c/IR.h"14#include "mlir-c/Support.h"15 16#ifdef __cplusplus17extern "C" {18#endif19 20MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(NVGPU, nvgpu);21 22//===---------------------------------------------------------------------===//23// TensorMapDescriptorType24//===---------------------------------------------------------------------===//25 26MLIR_CAPI_EXPORTED bool mlirTypeIsANVGPUTensorMapDescriptorType(MlirType type);27 28MLIR_CAPI_EXPORTED MlirType mlirNVGPUTensorMapDescriptorTypeGet(29 MlirContext ctx, MlirType tensorMemrefType, int swizzle, int l2promo,30 int oobFill, int interleave);31 32#ifdef __cplusplus33}34#endif35 36#endif // MLIR_C_DIALECT_NVGPU_H37