[llvm] [LV] Add test for WidenCall with mixed scalar-vector operands (PR #203092)

[Ramkumar Ramachandra via llvm-commits]

https://github.com/artagnon updated https://github.com/llvm/llvm-project/pull/203092

>From 5478576f1d66658dbb170b3380fe6a974213ca54 Mon Sep 17 00:00:00 2001
From: Ramkumar Ramachandra <artagnon at tenstorrent.com>
Date: Mon, 15 Jun 2026 10:03:26 +0100
Subject: [PATCH 1/2] [VPlan] Test for WideCall

---
 .../widen-call-op-scalar-vector.ll            | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll

diff --git a/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll b/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll
new file mode 100644
index 0000000000000..fc65a5c511f06
--- /dev/null
+++ b/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll
@@ -0,0 +1,46 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals none --version 6
+; RUN: opt -passes=loop-vectorize -force-vector-width=2 -force-vector-interleave=1 -S %s | FileCheck %s
+
+define void @test(ptr %p) {
+; CHECK-LABEL: define void @test(
+; CHECK-SAME: ptr [[P:%.*]]) {
+; CHECK-NEXT:  [[ENTRY:.*:]]
+; CHECK-NEXT:    br label %[[VECTOR_PH:.*]]
+; CHECK:       [[VECTOR_PH]]:
+; CHECK-NEXT:    br label %[[VECTOR_BODY:.*]]
+; CHECK:       [[VECTOR_BODY]]:
+; CHECK-NEXT:    [[INDEX:%.*]] = phi i64 [ 0, %[[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], %[[VECTOR_BODY]] ]
+; CHECK-NEXT:    [[VEC_IND:%.*]] = phi <2 x i64> [ <i64 0, i64 1>, %[[VECTOR_PH]] ], [ [[VEC_IND_NEXT:%.*]], %[[VECTOR_BODY]] ]
+; CHECK-NEXT:    [[TMP0:%.*]] = getelementptr i64, ptr [[P]], i64 [[INDEX]]
+; CHECK-NEXT:    [[TMP1:%.*]] = extractelement <2 x i64> [[VEC_IND]], i64 0
+; CHECK-NEXT:    [[TMP2:%.*]] = call <2 x i64> @vec_foo(<2 x i64> [[VEC_IND]], i64 [[TMP1]])
+; CHECK-NEXT:    store <2 x i64> [[TMP2]], ptr [[TMP0]], align 8
+; CHECK-NEXT:    [[INDEX_NEXT]] = add nuw i64 [[INDEX]], 2
+; CHECK-NEXT:    [[VEC_IND_NEXT]] = add <2 x i64> [[VEC_IND]], splat (i64 2)
+; CHECK-NEXT:    [[TMP3:%.*]] = icmp eq i64 [[INDEX_NEXT]], 100
+; CHECK-NEXT:    br i1 [[TMP3]], label %[[MIDDLE_BLOCK:.*]], label %[[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]]
+; CHECK:       [[MIDDLE_BLOCK]]:
+; CHECK-NEXT:    br label %[[EXIT:.*]]
+; CHECK:       [[EXIT]]:
+; CHECK-NEXT:    ret void
+;
+entry:
+  br label %loop
+
+loop:
+  %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
+  %gep = getelementptr i64, ptr %p, i64 %iv
+  %r = call i64 @foo(i64 %iv, i64 %iv) #0
+  store i64 %r, ptr %gep, align 8
+  %iv.next = add i64 %iv, 1
+  %ec = icmp eq i64 %iv.next, 100
+  br i1 %ec, label %exit, label %loop
+
+exit:
+  ret void
+}
+
+declare i64 @foo(i64, i64)
+declare <2 x i64> @vec_foo(<2 x i64>, i64)
+
+attributes #0 = { nounwind "vector-function-abi-variant"="_ZGVnN2vl1_foo(vec_foo)" }

>From bd49ddc116c33e1d42f2b08730b89a59f3ca2c4d Mon Sep 17 00:00:00 2001
From: Ramkumar Ramachandra <artagnon at tenstorrent.com>
Date: Fri, 19 Jun 2026 08:28:11 +0100
Subject: [PATCH 2/2] [LV] Add short comment for test

---
 .../Transforms/LoopVectorize/widen-call-op-scalar-vector.ll     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll b/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll
index fc65a5c511f06..1266a830417a8 100644
--- a/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll
+++ b/llvm/test/Transforms/LoopVectorize/widen-call-op-scalar-vector.ll
@@ -1,6 +1,8 @@
 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals none --version 6
 ; RUN: opt -passes=loop-vectorize -force-vector-width=2 -force-vector-interleave=1 -S %s | FileCheck %s
 
+; Test with call where same value is used as call argument in
+; vector and scalar position.
 define void @test(ptr %p) {
 ; CHECK-LABEL: define void @test(
 ; CHECK-SAME: ptr [[P:%.*]]) {