Add ArmSME type conversion unit test

Author: MacDue

mlir/unittests/Dialect/ArmSME/CMakeLists.txt

@@ -0,0 +1,5 @@
+add_mlir_unittest(MLIRArmSMETests
+  TileTypeConversionTest.cpp)
+target_link_libraries(MLIRArmSMETests
+  PRIVATE
+  MLIRArmSMEToLLVM)

mlir/unittests/Dialect/ArmSME/TileTypeConversionTest.cpp

@@ -0,0 +1,51 @@
+//===- TileTypeConversionTest.cpp - Tests ArmSME tile type conversion -----===//
+//
+// 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/Conversion/ArmSMEToLLVM/ArmSMEToLLVM.h"
+#include "mlir/Conversion/LLVMCommon/ConversionTarget.h"
+#include "mlir/Conversion/LLVMCommon/Pattern.h"
+#include "mlir/Conversion/LLVMCommon/TypeConverter.h"
+#include "mlir/Dialect/ArmSME/IR/ArmSME.h"
+
+#include "gtest/gtest.h"
+
+using namespace mlir;
+
+class ArmSMETest : public ::testing::Test {
+protected:
+  ArmSMETest() { context.getOrLoadDialect<mlir::arm_sme::ArmSMEDialect>(); }
+
+  mlir::MLIRContext context;
+};
+
+TEST_F(ArmSMETest, TestTileTypeConversion) {
+  LLVMTypeConverter llvmConverer(&context);
+  LLVMTypeConverter llvmConvererWithArmSMEConversion(&context);
+
+  RewritePatternSet patterns(&context);
+  populateArmSMEToLLVMConversionPatterns(llvmConvererWithArmSMEConversion,
+                                         patterns);
+
+  Type i32 = IntegerType::get(&context, 32);
+  auto smeTileType = VectorType::get({4, 4}, i32, {true, true});
+
+  // An unmodified LLVMTypeConverer should fail to convert an ArmSME tile type.
+  {
+    SmallVector<Type> convertedType;
+    ASSERT_TRUE(failed(llvmConverer.convertType(smeTileType, convertedType)));
+  }
+
+  // An updated LLVMTypeConverer should return the ArmSME tile vector type
+  // unchanged.
+  {
+    SmallVector<Type> convertedType;
+    ASSERT_TRUE(succeeded(llvmConvererWithArmSMEConversion.convertType(
+        smeTileType, convertedType)));
+    ASSERT_EQ(ArrayRef<Type>(convertedType), ArrayRef<Type>{smeTileType});
+  }
+}

mlir/unittests/Dialect/CMakeLists.txt

@@ -6,6 +6,7 @@ target_link_libraries(MLIRDialectTests
   MLIRIR
   MLIRDialect)
 
+add_subdirectory(ArmSME)
 add_subdirectory(Index)
 add_subdirectory(LLVMIR)
 add_subdirectory(MemRef)