[llvm] [GlobalISel] Add unit tests for call lowering on byref support (PR #96805)

[via llvm-commits]

https://github.com/darkbuck created https://github.com/llvm/llvm-project/pull/96805

None

>From a420bb98e94c60aa97d3584263db6cc61763032a Mon Sep 17 00:00:00 2001
From: Michael Liao <michael.hliao at gmail.com>
Date: Wed, 26 Jun 2024 14:14:46 -0400
Subject: [PATCH] =?UTF-8?q?[=F0=9D=98=80=F0=9D=97=BD=F0=9D=97=BF]=20initia?=
 =?UTF-8?q?l=20version?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Created using spr 1.3.4
---
 .../CodeGen/GlobalISel/CMakeLists.txt         |  2 +
 .../CodeGen/GlobalISel/CallLowering.cpp       | 57 +++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 llvm/unittests/CodeGen/GlobalISel/CallLowering.cpp

diff --git a/llvm/unittests/CodeGen/GlobalISel/CMakeLists.txt b/llvm/unittests/CodeGen/GlobalISel/CMakeLists.txt
index 111d7f4d2f620..253b90d5dc1f8 100644
--- a/llvm/unittests/CodeGen/GlobalISel/CMakeLists.txt
+++ b/llvm/unittests/CodeGen/GlobalISel/CMakeLists.txt
@@ -1,5 +1,6 @@
 set(LLVM_LINK_COMPONENTS
   ${LLVM_TARGETS_TO_BUILD}
+  AsmParser
   CodeGen
   CodeGenTypes
   Core
@@ -26,4 +27,5 @@ add_llvm_unittest(GlobalISelTests
   KnownBitsVectorTest.cpp
   GISelUtilsTest.cpp
   GISelAliasTest.cpp
+  CallLowering.cpp
   )
diff --git a/llvm/unittests/CodeGen/GlobalISel/CallLowering.cpp b/llvm/unittests/CodeGen/GlobalISel/CallLowering.cpp
new file mode 100644
index 0000000000000..1d36e5e9ca946
--- /dev/null
+++ b/llvm/unittests/CodeGen/GlobalISel/CallLowering.cpp
@@ -0,0 +1,57 @@
+//===- CallLowering.cpp - CallLowering 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/CodeGen/GlobalISel/CallLowering.h"
+#include "llvm/ADT/StringRef.h"
+#include "llvm/AsmParser/Parser.h"
+#include "llvm/IR/LLVMContext.h"
+#include "llvm/IR/Module.h"
+#include "llvm/Support/SourceMgr.h"
+#include "gtest/gtest.h"
+
+using namespace llvm;
+
+namespace {
+struct TargetCallLoweringTest : public CallLowering, testing::Test {
+  LLVMContext C;
+
+public:
+  TargetCallLoweringTest() : CallLowering(nullptr) {}
+
+  std::unique_ptr<Module> parseIR(const char *IR) {
+    SMDiagnostic Err;
+    std::unique_ptr<Module> Mod = parseAssemblyString(IR, Err, C);
+    if (!Mod)
+      Err.print("TestTargetCallLoweringTest", errs());
+    return Mod;
+  }
+};
+} // namespace
+
+TEST_F(TargetCallLoweringTest, ArgByRef) {
+  std::unique_ptr<Module> M = parseIR(R"(
+define void @foo(ptr %p0, ptr byref(i32) align(4) %p1) {
+  ret void
+}
+)");
+  const DataLayout &DL = M->getDataLayout();
+  Function *F = M->getFunction("foo");
+  // Dummy vregs.
+  SmallVector<Register, 4> VRegs(1, 1);
+
+  CallLowering::ArgInfo Arg0(VRegs, F->getArg(0)->getType(), 0);
+  setArgFlags(Arg0, AttributeList::FirstArgIndex + 0, DL, *F);
+  EXPECT_TRUE(Arg0.Flags[0].isPointer());
+  EXPECT_FALSE(Arg0.Flags[0].isByRef());
+
+  CallLowering::ArgInfo Arg1(VRegs, F->getArg(1)->getType(), 1);
+  setArgFlags(Arg1, AttributeList::FirstArgIndex + 1, DL, *F);
+  EXPECT_TRUE(Arg1.Flags[0].isPointer());
+  EXPECT_TRUE(Arg1.Flags[0].isByRef());
+  EXPECT_EQ(Arg1.Flags[0].getByRefSize(), 4U);
+}