[llvm] 1146d40 - [UnitTests] Add ShuffleVectorInst unit test coverage for shuffle mask kind matchers

[Simon Pilgrim via llvm-commits]

Author: Simon Pilgrim
Date: 2022-09-19T11:53:30+01:00
New Revision: 1146d40d9ab2edaa45704db44bb1479a54dd3fcc

URL: https://github.com/llvm/llvm-project/commit/1146d40d9ab2edaa45704db44bb1479a54dd3fcc
DIFF: https://github.com/llvm/llvm-project/commit/1146d40d9ab2edaa45704db44bb1479a54dd3fcc.diff

LOG: [UnitTests] Add ShuffleVectorInst unit test coverage for shuffle mask kind matchers

Add tests for the core static shuffle pattern match helpers

Added: 
    llvm/unittests/IR/ShuffleVectorInstTest.cpp

Modified: 
    llvm/unittests/IR/CMakeLists.txt

Removed: 
    


################################################################################
diff  --git a/llvm/unittests/IR/CMakeLists.txt b/llvm/unittests/IR/CMakeLists.txt
index e9f3d5598d142..2ce8596446d62 100644
--- a/llvm/unittests/IR/CMakeLists.txt
+++ b/llvm/unittests/IR/CMakeLists.txt
@@ -34,6 +34,7 @@ add_llvm_unittest(IRTests
   ModuleTest.cpp
   PassManagerTest.cpp
   PatternMatch.cpp
+  ShuffleVectorInstTest.cpp
   TimePassesTest.cpp
   TypesTest.cpp
   UseTest.cpp

diff  --git a/llvm/unittests/IR/ShuffleVectorInstTest.cpp b/llvm/unittests/IR/ShuffleVectorInstTest.cpp
new file mode 100644
index 0000000000000..cf4f91cd2a84d
--- /dev/null
+++ b/llvm/unittests/IR/ShuffleVectorInstTest.cpp
@@ -0,0 +1,119 @@
+//===- llvm/unittest/IR/ShuffleVectorInstTest.cpp - Shuffle unit tests ----===//
+//
+// 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 "llvm/IR/Instructions.h"
+#include "gtest/gtest.h"
+
+using namespace llvm;
+
+namespace {
+
+TEST(ShuffleVectorInst, isIdentityMask) {
+  ASSERT_TRUE(ShuffleVectorInst::isIdentityMask({0, 1, 2, 3}));
+  ASSERT_TRUE(ShuffleVectorInst::isIdentityMask({0, 1, 2, 3, -1}));
+  ASSERT_TRUE(ShuffleVectorInst::isIdentityMask({0, 1, -1, 3}));
+
+  ASSERT_FALSE(ShuffleVectorInst::isIdentityMask({0, 1, 2, 4}));
+  ASSERT_FALSE(ShuffleVectorInst::isIdentityMask({0, -1, 2, 4}));
+}
+
+TEST(ShuffleVectorInst, isSelectMask) {
+  ASSERT_TRUE(ShuffleVectorInst::isSelectMask({0, 5, 6, 3}));
+
+  ASSERT_FALSE(ShuffleVectorInst::isSelectMask({0, 1, 2, 3}));
+}
+
+TEST(ShuffleVectorInst, isReverseMask) {
+  ASSERT_TRUE(ShuffleVectorInst::isReverseMask({3, 2, 1, 0}));
+  ASSERT_TRUE(ShuffleVectorInst::isReverseMask({-1, -1, 1, 0}));
+
+  ASSERT_FALSE(ShuffleVectorInst::isReverseMask({4, 3, 2, 1}));
+}
+
+TEST(ShuffleVectorInst, isZeroEltSplatMask) {
+  ASSERT_TRUE(ShuffleVectorInst::isZeroEltSplatMask({0, 0, 0, 0}));
+  ASSERT_TRUE(ShuffleVectorInst::isZeroEltSplatMask({0, -1, 0, -1}));
+
+  ASSERT_FALSE(ShuffleVectorInst::isZeroEltSplatMask({1, 1, 1, 1}));
+}
+
+TEST(ShuffleVectorInst, isTransposeMask) {
+  ASSERT_TRUE(ShuffleVectorInst::isTransposeMask({0, 4, 2, 6}));
+  ASSERT_TRUE(ShuffleVectorInst::isTransposeMask({1, 5, 3, 7}));
+
+  ASSERT_FALSE(ShuffleVectorInst::isTransposeMask({2, 6, 4, 8}));
+}
+
+TEST(ShuffleVectorInst, isSpliceMask) {
+  int Index;
+
+  ASSERT_TRUE(ShuffleVectorInst::isSpliceMask({0, 1, 2, 3}, Index));
+  ASSERT_EQ(0, Index);
+
+  ASSERT_TRUE(ShuffleVectorInst::isSpliceMask({1, 2, 3, 4, 5, 6, 7}, Index));
+  ASSERT_EQ(1, Index);
+
+  ASSERT_FALSE(ShuffleVectorInst::isSpliceMask({4, 5, 6, 7}, Index));
+}
+
+TEST(ShuffleVectorInst, isExtractSubvectorMask) {
+  int Index;
+
+  ASSERT_TRUE(
+      ShuffleVectorInst::isExtractSubvectorMask({0, 1, 2, 3}, 8, Index));
+  ASSERT_EQ(0, Index);
+
+  ASSERT_TRUE(
+      ShuffleVectorInst::isExtractSubvectorMask({-1, 3, 4, 5}, 8, Index));
+  ASSERT_EQ(2, Index);
+
+  ASSERT_FALSE(
+      ShuffleVectorInst::isExtractSubvectorMask({1, 2, 3, -1}, 4, Index));
+}
+
+TEST(ShuffleVectorInst, isInsertSubvectorMask) {
+  int NumSubElts, Index;
+
+  ASSERT_TRUE(ShuffleVectorInst::isInsertSubvectorMask(
+      {8, 9, 10, 11, 4, 5, 6, 7}, 8, NumSubElts, Index));
+  ASSERT_EQ(0, Index);
+  ASSERT_EQ(4, NumSubElts);
+
+  ASSERT_TRUE(
+      ShuffleVectorInst::isInsertSubvectorMask({0, 2}, 2, NumSubElts, Index));
+  ASSERT_EQ(1, Index);
+  ASSERT_EQ(1, NumSubElts);
+}
+
+TEST(ShuffleVectorInst, isReplicationMask) {
+  int ReplicationFactor, VF;
+
+  ASSERT_TRUE(ShuffleVectorInst::isReplicationMask({0, 0, 1, 1, 2, 2, 3, 3},
+                                                   ReplicationFactor, VF));
+  ASSERT_EQ(2, ReplicationFactor);
+  ASSERT_EQ(4, VF);
+
+  ASSERT_TRUE(ShuffleVectorInst::isReplicationMask(
+      {0, 0, 0, 1, 1, 1, -1, -1, -1, 3, 3, 3, 4, 4, 4}, ReplicationFactor, VF));
+  ASSERT_EQ(3, ReplicationFactor);
+  ASSERT_EQ(5, VF);
+}
+
+TEST(ShuffleVectorInst, isOneUseSingleSourceMask) {
+  ASSERT_TRUE(
+      ShuffleVectorInst::isOneUseSingleSourceMask({0, 1, 2, 3, 3, 2, 0, 1}, 4));
+  ASSERT_TRUE(
+      ShuffleVectorInst::isOneUseSingleSourceMask({2, 3, 4, 5, 6, 7, 0, 1}, 8));
+
+  ASSERT_FALSE(ShuffleVectorInst::isOneUseSingleSourceMask(
+      {0, -1, 2, 3, 3, 2, 0, 1}, 4));
+  ASSERT_FALSE(
+      ShuffleVectorInst::isOneUseSingleSourceMask({0, 1, 2, 3, 3, 3, 1, 0}, 4));
+}
+
+} // end anonymous namespace