mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 10:56:07 +00:00
f42e1a7484
This is a rollforward with fixes of https://reviews.llvm.org/rGbcc10817d5569172ee065015747e226280e9b698 (originally https://reviews.llvm.org/D142394). The original patch exposed an asan problem on aarch64, which is fixed by simply calling the context destructors properly. Reviewed By: rriddle Differential Revision: https://reviews.llvm.org/D143294
90 lines
3.0 KiB
C
90 lines
3.0 KiB
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;
|
|
}
|