73 lines · c
1//===-- mlir-c/Dialect/GPU.h - C API for GPU 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_GPU_H11#define MLIR_C_DIALECT_GPU_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(GPU, gpu);21 22//===-------------------------------------------------------------------===//23// AsyncTokenType24//===-------------------------------------------------------------------===//25 26MLIR_CAPI_EXPORTED bool mlirTypeIsAGPUAsyncTokenType(MlirType type);27 28MLIR_CAPI_EXPORTED MlirType mlirGPUAsyncTokenTypeGet(MlirContext ctx);29 30//===---------------------------------------------------------------------===//31// ObjectAttr32//===---------------------------------------------------------------------===//33 34MLIR_CAPI_EXPORTED bool mlirAttributeIsAGPUObjectAttr(MlirAttribute attr);35 36MLIR_CAPI_EXPORTED MlirAttribute37mlirGPUObjectAttrGet(MlirContext mlirCtx, MlirAttribute target, uint32_t format,38 MlirStringRef objectStrRef, MlirAttribute mlirObjectProps);39 40MLIR_CAPI_EXPORTED MlirAttribute mlirGPUObjectAttrGetWithKernels(41 MlirContext mlirCtx, MlirAttribute target, uint32_t format,42 MlirStringRef objectStrRef, MlirAttribute mlirObjectProps,43 MlirAttribute mlirKernelsAttr);44 45MLIR_CAPI_EXPORTED MlirAttribute46mlirGPUObjectAttrGetTarget(MlirAttribute mlirObjectAttr);47 48MLIR_CAPI_EXPORTED uint32_t49mlirGPUObjectAttrGetFormat(MlirAttribute mlirObjectAttr);50 51MLIR_CAPI_EXPORTED MlirStringRef52mlirGPUObjectAttrGetObject(MlirAttribute mlirObjectAttr);53 54MLIR_CAPI_EXPORTED bool55mlirGPUObjectAttrHasProperties(MlirAttribute mlirObjectAttr);56 57MLIR_CAPI_EXPORTED MlirAttribute58mlirGPUObjectAttrGetProperties(MlirAttribute mlirObjectAttr);59 60MLIR_CAPI_EXPORTED bool61mlirGPUObjectAttrHasKernels(MlirAttribute mlirObjectAttr);62 63MLIR_CAPI_EXPORTED MlirAttribute64mlirGPUObjectAttrGetKernels(MlirAttribute mlirObjectAttr);65 66#ifdef __cplusplus67}68#endif69 70#include "mlir/Dialect/GPU/Transforms/Passes.capi.h.inc"71 72#endif // MLIR_C_DIALECT_GPU_H73