[flang-commits] [flang] [flang] Catch attempts to use assumed-rank as elemental argument (PR #159852)

[Peter Klausler via flang-commits]

https://github.com/klausler created https://github.com/llvm/llvm-project/pull/159852

An assumed-rank array may not be used as an argument to an elemental procedure.

Fixes https://github.com/llvm/llvm-project/issues/159555.

>From 1c0e09b8ab39210f2ceb01e39c24a60bb9425c91 Mon Sep 17 00:00:00 2001
From: Peter Klausler <pklausler at nvidia.com>
Date: Fri, 19 Sep 2025 14:38:55 -0700
Subject: [PATCH] [flang] Catch attempts to use assumed-rank as elemental
 argument

An assumed-rank array may not be used as an argument to an
elemental procedure.

Fixes https://github.com/llvm/llvm-project/issues/159555.
---
 flang/lib/Semantics/check-call.cpp   |  4 ++++
 flang/test/Semantics/elemental03.f90 | 13 +++++++++++++
 2 files changed, 17 insertions(+)
 create mode 100644 flang/test/Semantics/elemental03.f90

diff --git a/flang/lib/Semantics/check-call.cpp b/flang/lib/Semantics/check-call.cpp
index f0078fda3600c..f290682550a61 100644
--- a/flang/lib/Semantics/check-call.cpp
+++ b/flang/lib/Semantics/check-call.cpp
@@ -1483,6 +1483,10 @@ static bool CheckElementalConformance(parser::ContextualMessages &messages,
             evaluate::SayWithDeclaration(messages, *wholeSymbol,
                 "Whole assumed-size array '%s' may not be used as an argument to an elemental procedure"_err_en_US,
                 wholeSymbol->name());
+          } else if (IsAssumedRank(*wholeSymbol)) {
+            evaluate::SayWithDeclaration(messages, *wholeSymbol,
+                "Assumed-rank array '%s' may not be used as an argument to an elemental procedure"_err_en_US,
+                wholeSymbol->name());
           }
         }
         if (auto argShape{evaluate::GetShape(context, *expr)}) {
diff --git a/flang/test/Semantics/elemental03.f90 b/flang/test/Semantics/elemental03.f90
new file mode 100644
index 0000000000000..1a2e22065b418
--- /dev/null
+++ b/flang/test/Semantics/elemental03.f90
@@ -0,0 +1,13 @@
+!RUN: %python %S/test_errors.py %s %flang_fc1
+module m
+ contains
+  elemental real function f(x)
+    real, intent(in) :: x
+    f = x
+  end
+  subroutine s(a)
+    real a(..)
+    !ERROR: Assumed-rank array 'a' may not be used as an argument to an elemental procedure
+    print *, f(a)
+  end
+end