TheRock/compiler/amd-llvm/mlir/test/CAPI/transform.c

//===- transform.c - Test of Transform dialect C API ----------------------===//
//
// 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-transform-test 2>&1 | FileCheck %s

#include "mlir-c/Dialect/Transform.h"
#include "mlir-c/IR.h"
#include "mlir-c/Support.h"

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>

// CHECK-LABEL: testAnyOpType
void testAnyOpType(MlirContext ctx) {
  fprintf(stderr, "testAnyOpType\n");

  MlirType parsedType = mlirTypeParseGet(
      ctx, mlirStringRefCreateFromCString("!transform.any_op"));
  MlirType constructedType = mlirTransformAnyOpTypeGet(ctx);

  assert(!mlirTypeIsNull(parsedType) && "couldn't parse AnyOpType");
  assert(!mlirTypeIsNull(constructedType) && "couldn't construct AnyOpType");

  // CHECK: equal: 1
  fprintf(stderr, "equal: %d\n", mlirTypeEqual(parsedType, constructedType));

  // CHECK: parsedType isa AnyOpType: 1
  fprintf(stderr, "parsedType isa AnyOpType: %d\n",
          mlirTypeIsATransformAnyOpType(parsedType));
  // CHECK: parsedType isa OperationType: 0
  fprintf(stderr, "parsedType isa OperationType: %d\n",
          mlirTypeIsATransformOperationType(parsedType));

  // CHECK: !transform.any_op
  mlirTypeDump(constructedType);

  fprintf(stderr, "\n\n");
}

// CHECK-LABEL: testOperationType
void testOperationType(MlirContext ctx) {
  fprintf(stderr, "testOperationType\n");

  MlirType parsedType = mlirTypeParseGet(
      ctx, mlirStringRefCreateFromCString("!transform.op<\"foo.bar\">"));
  MlirType constructedType = mlirTransformOperationTypeGet(
      ctx, mlirStringRefCreateFromCString("foo.bar"));

  assert(!mlirTypeIsNull(parsedType) && "couldn't parse AnyOpType");
  assert(!mlirTypeIsNull(constructedType) && "couldn't construct AnyOpType");

  // CHECK: equal: 1
  fprintf(stderr, "equal: %d\n", mlirTypeEqual(parsedType, constructedType));

  // CHECK: parsedType isa AnyOpType: 0
  fprintf(stderr, "parsedType isa AnyOpType: %d\n",
          mlirTypeIsATransformAnyOpType(parsedType));
  // CHECK: parsedType isa OperationType: 1
  fprintf(stderr, "parsedType isa OperationType: %d\n",
          mlirTypeIsATransformOperationType(parsedType));

  // CHECK: operation name equal: 1
  MlirStringRef operationName =
      mlirTransformOperationTypeGetOperationName(constructedType);
  fprintf(stderr, "operation name equal: %d\n",
          mlirStringRefEqual(operationName,
                             mlirStringRefCreateFromCString("foo.bar")));

  // CHECK: !transform.op<"foo.bar">
  mlirTypeDump(constructedType);

  fprintf(stderr, "\n\n");
}

int main(void) {
  MlirContext ctx = mlirContextCreate();
  mlirDialectHandleRegisterDialect(mlirGetDialectHandle__transform__(), ctx);
  testAnyOpType(ctx);
  testOperationType(ctx);
  mlirContextDestroy(ctx);
  return EXIT_SUCCESS;
}