Precommit basic unit test for visitors

Author: jasilvanus

test/unit/dialect/TestDialect.td

@@ -135,3 +135,26 @@ def DefaultNameTargetExtType : DialectType<TestDialect, "default.target"> {
     Type name should be "test.default.target".
   }];
 }
+
+def SomeBaseOpClass : OpClass<TestDialect> {
+  let arguments = (ins);
+  let summary = "family of operations";
+  let description = [{
+    Illustrate the use of the OpClass feature.
+  }];
+}
+
+class ConcreteOpClassMemberOp<string op>
+    : TestOp<"some.op.class.subop." # op, []> {
+  let superclass = SomeBaseOpClass;
+  let results = (outs);
+  let arguments = (ins superclass);
+
+  let summary = "perform the " # op # " operation";
+  let description = [{
+    Illustrate the use of the OpClass feature.
+  }];
+}
+
+def AddOp : ConcreteOpClassMemberOp<"add">;
+def MulOp : ConcreteOpClassMemberOp<"mul">;

test/unit/interface/CMakeLists.txt

@@ -21,6 +21,7 @@ add_dialects_unit_test(DialectsADTTests
     OpSetTests.cpp
     OpMapTests.cpp
     OpMapIRTests.cpp
-    DialectTypeTests.cpp)
+    DialectTypeTests.cpp
+    VisitorIRTests.cpp)
 
 add_dependencies(DialectsADTTests TestDialectTableGen)

test/unit/interface/VisitorIRTests.cpp

@@ -0,0 +1,87 @@
+/*
+ ***********************************************************************************************************************
+ * Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ ***********************************************************************************************************************
+ */
+
+#include "TestDialect.h"
+#include "llvm-dialects/Dialect/Builder.h"
+#include "llvm-dialects/Dialect/Dialect.h"
+#include "llvm-dialects/Dialect/Visitor.h"
+#include "llvm/ADT/StringRef.h"
+#include "llvm/IR/BasicBlock.h"
+#include "llvm/IR/Function.h"
+#include "llvm/IR/Intrinsics.h"
+#include "llvm/IR/Module.h"
+#include "gtest/gtest.h"
+
+#include <memory>
+
+using namespace llvm;
+using namespace llvm_dialects;
+
+class VisitorIRTestFixture : public testing::Test {
+protected:
+  void SetUp() override {
+    setupDialectsContext();
+    makeModule();
+  }
+
+  LLVMContext Context;
+  std::unique_ptr<DialectContext> DC;
+  std::unique_ptr<Module> Mod;
+  Function *EP = nullptr;
+
+  BasicBlock *getEntryBlock() { return EntryBlock; }
+
+private:
+  BasicBlock *EntryBlock = nullptr;
+
+  void makeModule() {
+    Mod = std::make_unique<Module>("dialects_test", Context);
+    const std::array<Type *, 1> Args = {Type::getInt32Ty(Mod->getContext())};
+    FunctionCallee FC = Mod->getOrInsertFunction(
+        "main",
+        FunctionType::get(Type::getVoidTy(Mod->getContext()), Args, false));
+    EP = cast<Function>(FC.getCallee());
+    EntryBlock = BasicBlock::Create(Mod->getContext(), "entry", EP);
+  }
+
+  void setupDialectsContext() {
+    DC = DialectContext::make<test::TestDialect>(Context);
+  }
+};
+
+TEST_F(VisitorIRTestFixture, VisitOp) {
+  llvm_dialects::Builder Builder{Context};
+  Builder.SetInsertPoint(getEntryBlock());
+
+  auto *Mul1 = Builder.create<test::MulOp>();
+  auto *Mul2 = Builder.create<test::MulOp>();
+  auto *Add1 = Builder.create<test::AddOp>();
+  auto *Add2 = Builder.create<test::AddOp>();
+  auto *DialectOp1 = Builder.create<test::DialectOp1>();
+
+  DenseSet<test::MulOp *> Ops;
+  static const auto Visitor =
+      llvm_dialects::VisitorBuilder<DenseSet<test::MulOp *>>()
+          .add<test::MulOp>([](auto &Ops, test::MulOp &Op) { Ops.insert(&Op); })
+          .build();
+  Visitor.visit(Ops, *Mod);
+  EXPECT_EQ(Ops.size(), 2);
+  for (const auto *Op : Ops) {
+    EXPECT_TRUE(Op == Mul1 || Op == Mul2);
+  }
+}