From 3fb8e313a877a1267948376f59e6dfaf1ea3c957 Mon Sep 17 00:00:00 2001 From: Joren Dumoulin Date: Tue, 17 Dec 2024 15:54:56 +0100 Subject: [PATCH] add aie-opt lit testing config --- tests/filecheck/dialects/aie.mlir | 5 +++-- tests/filecheck/lit.cfg | 8 ++++++++ 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/tests/filecheck/dialects/aie.mlir b/tests/filecheck/dialects/aie.mlir index a1db132..4268f12 100644 --- a/tests/filecheck/dialects/aie.mlir +++ b/tests/filecheck/dialects/aie.mlir @@ -1,5 +1,6 @@ // RUN: XDSL_ROUNDTRIP +// RUN: AIE_ROUNDTRIP -"test.op"() : () -> () +arith.constant 5 : i32 -// CHECK: "test.op"() : () -> () +// CHECK: arith.constant 5 : i32 diff --git a/tests/filecheck/lit.cfg b/tests/filecheck/lit.cfg index 09d148e..a07e1a6 100644 --- a/tests/filecheck/lit.cfg +++ b/tests/filecheck/lit.cfg @@ -1,5 +1,6 @@ import lit.formats import os +import shutil config.name = "xdsl-aie" config.test_format = lit.formats.ShTest() @@ -7,3 +8,10 @@ config.suffixes = ['.mlir'] config.substitutions.append(('XDSL_ROUNDTRIP', "xdsl-aie-opt %s --print-op-generic --split-input-file | xdsl-aie-opt --split-input-file | filecheck %s")) config.substitutions.append(("XDSL_GENERIC_ROUNDTRIP", "xdsl-aie-opt %s --print-op-generic --split-input-file | filecheck %s --check-prefix=CHECK-GENERIC")) + +if shutil.which("aie-opt"): + config.substitutions.append(('AIE_ROUNDTRIP', "xdsl-aie-opt %s --split-input-file | aie-opt --split-input-file | xdsl-aie-opt --split-input-file | filecheck %s")) + config.substitutions.append(("AIE_GENERIC_ROUNDTRIP", "xdsl-aie-opt %s --print-op-generic --split-input-file | aie-opt --split-input-file --mlir-print-op-generic | xdsl-aie-opt --split-input-file --print-op-generic | filecheck %s --check-prefix=CHECK-GENERIC")) +else: + config.substitutions.append(('AIE_ROUNDTRIP', ":")) + config.substitutions.append(("AIE_GENERIC_ROUNDTRIP", ":"))