Add C API files for the LLVM dialect

[apaszke]

Summary

For now only expose a builder for the LLVM pointer type.

[ftynse]

I suppose this also needs cmake changes.

mlir/include/mlir-c/Dialect/LLVM.h

23

mlir/lib/CAPI/Dialect/LLVM.cpp

2

[apaszke]

Update CMake configuration, added tests

[apaszke]

Addressed review comments

mlir/include/mlir-c/Dialect/LLVM.h

+//===-- mlir-c/Dialect/LLVM.h - C API for LLVM --------------------*- C -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM
+// Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef MLIR_C_DIALECT_LLVM_H
+#define MLIR_C_DIALECT_LLVM_H
+
+#include "mlir-c/IR.h"
+#include "mlir-c/Registration.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(LLVM, llvm);
+
+/// Creates an llvm.ptr type.
+MLIR_CAPI_EXPORTED MlirType mlirLLVMPointerTypeGet(MlirType pointee,
+                                                   unsigned addressSpace);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif // MLIR_C_DIALECT_LLVM_H

mlir/lib/CAPI/Dialect/CMakeLists.txt

add_mlir_public_c_api_library(MLIRCAPIGPU
   MLIRGPUPassIncGen
 
   LINK_LIBS PUBLIC
   MLIRCAPIIR
   MLIRGPU
   MLIRPass
 )
 
+add_mlir_public_c_api_library(MLIRCAPILLVM
+  LLVM.cpp
+
+  PARTIAL_SOURCES_INTENDED
+  LINK_LIBS PUBLIC
+  MLIRCAPIIR
+  MLIRLLVMIR
+)
+
 add_mlir_public_c_api_library(MLIRCAPILinalg
   Linalg.cpp
   LinalgPasses.cpp
 
   PARTIAL_SOURCES_INTENDED
   DEPENDS
   MLIRLinalgPassIncGen
 

mlir/lib/CAPI/Dialect/LLVM.cpp

+//===- LLVM.cpp - C Interface for LLVM dialect ----------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "mlir-c/Dialect/LLVM.h"
+#include "mlir/CAPI/Registration.h"
+#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
+#include "mlir/Dialect/LLVMIR/LLVMTypes.h"
+
+using namespace mlir;
+using namespace mlir::LLVM;
+
+MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(LLVM, llvm, LLVMDialect)
+
+MlirType mlirLLVMPointerTypeGet(MlirType pointee, unsigned addressSpace) {
+  return wrap(LLVMPointerType::get(unwrap(pointee), addressSpace));
+}

mlir/test/CAPI/CMakeLists.txt

 DEPENDS
   MLIRConversionPassIncGen
 )
 
 _add_capi_test_executable(mlir-capi-ir-test
   ir.c
 )
 
+_add_capi_test_executable(mlir-capi-llvm-test
+  llvm.c
+)
+
 _add_capi_test_executable(mlir-capi-pass-test
   pass.c
 )
 
 _add_capi_test_executable(mlir-capi-sparse-tensor-test
   sparse_tensor.c
 )

mlir/test/CAPI/llvm.c

+//===- llvm.c - Test of llvm APIs -----------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM
+// Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+// RUN: mlir-capi-llvm-test 2>&1 | FileCheck %s
+
+#include "mlir-c/Dialect/LLVM.h"
+#include "mlir-c/IR.h"
+#include "mlir-c/BuiltinTypes.h"
+
+#include <assert.h>
+#include <math.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+// CHECK-LABEL: testTypeCreation()
+static void testTypeCreation(MlirContext ctx) {
+  fprintf(stderr, "testTypeCreation()\n");
+  MlirType i32 = mlirIntegerTypeGet(ctx, 32);
+
+  const char *i32p_text = "!llvm.ptr<i32>";
+  MlirType i32p = mlirLLVMPointerTypeGet(i32, 0);
+  MlirType i32p_ref = mlirTypeParseGet(ctx, mlirStringRefCreateFromCString(i32p_text));
+  // CHECK: !llvm.ptr<i32>: 1
+  fprintf(stderr, "%s: %d\n", i32p_text, mlirTypeEqual(i32p, i32p_ref));
+
+  const char *i32p4_text = "!llvm.ptr<i32, 4>";
+  MlirType i32p4 = mlirLLVMPointerTypeGet(i32, 4);
+  MlirType i32p4_ref = mlirTypeParseGet(ctx, mlirStringRefCreateFromCString(i32p4_text));
+  // CHECK: !llvm.ptr<i32, 4>: 1
+  fprintf(stderr, "%s: %d\n", i32p4_text, mlirTypeEqual(i32p4, i32p4_ref));
+}
+
+int main() {
+  MlirContext ctx = mlirContextCreate();
+  mlirDialectHandleRegisterDialect(mlirGetDialectHandle__llvm__(), ctx);
+  mlirContextGetOrLoadDialect(ctx, mlirStringRefCreateFromCString("llvm"));
+  testTypeCreation(ctx);
+  mlirContextDestroy(ctx);
+  return 0;
+}
+

mlir/test/CMakeLists.txt

configure_lit_site_cfg(
   MAIN_CONFIG
   ${CMAKE_CURRENT_SOURCE_DIR}/Unit/lit.cfg.py
   )
 
 set(MLIR_TEST_DEPENDS
   FileCheck count not
   mlir-capi-execution-engine-test
   mlir-capi-ir-test
+  mlir-capi-llvm-test
   mlir-capi-pass-test
   mlir-capi-sparse-tensor-test
   mlir-cpu-runner
   mlir-linalg-ods-gen
   mlir-linalg-ods-yaml-gen
   mlir-lsp-server
   mlir-opt
   mlir-reduce