From 7257032c6445ad05d6ff5fb8908d4877a4e3c9ff Mon Sep 17 00:00:00 2001
From: Ron Lieberman <ron.lieberman@amd.com>
Date: Wed, 29 Apr 2026 11:42:25 -0500
Subject: [PATCH] [flang-288613*] add missing distribute, force malloc to heap
 on device

---
 test/smoke/flang-288613-2/Makefile         | 2 +-
 test/smoke/flang-288613-2/flang-288613.f90 | 4 ++--
 test/smoke/flang-288613-3/Makefile         | 2 +-
 test/smoke/flang-288613/Makefile           | 2 +-
 test/smoke/flang-288613/flang-288613.f90   | 4 ++--
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/test/smoke/flang-288613-2/Makefile b/test/smoke/flang-288613-2/Makefile
index 2c1d032424..25f5c5f444 100644
--- a/test/smoke/flang-288613-2/Makefile
+++ b/test/smoke/flang-288613-2/Makefile
@@ -7,7 +7,7 @@ TESTSRC_ALL  = $(TESTSRC_MAIN) $(TESTSRC_AUX)
 
 FLANG        ?= flang
 OMP_BIN      = $(AOMP)/bin/$(FLANG)
-CC           = $(OMP_BIN) $(VERBOSE)
+CC           = $(OMP_BIN) $(VERBOSE) -mmlir --enable-gpu-heap-alloc
 #-ccc-print-phases
 #"-\#\#\#"
 
diff --git a/test/smoke/flang-288613-2/flang-288613.f90 b/test/smoke/flang-288613-2/flang-288613.f90
index d52539e54a..8a7b9d17e4 100644
--- a/test/smoke/flang-288613-2/flang-288613.f90
+++ b/test/smoke/flang-288613-2/flang-288613.f90
@@ -16,7 +16,7 @@ end subroutine dec_val_dev
         integer i,j,k,l,m
         real(rstd), dimension(nsize,nsize,nsize) :: tmp
 
-!$omp target teams
+!$omp target teams distribute
         do i=1,nsize
 #define PRIVATE_ARRAY 1
 #ifdef PRIVATE_ARRAY
@@ -35,7 +35,7 @@ end subroutine dec_val_dev
                 end do
             end do
         end do
-!$omp end target teams
+!$omp end target teams distribute
     end subroutine _compute_dev
 
     subroutine compute_dev()
diff --git a/test/smoke/flang-288613-3/Makefile b/test/smoke/flang-288613-3/Makefile
index 2c1d032424..25f5c5f444 100644
--- a/test/smoke/flang-288613-3/Makefile
+++ b/test/smoke/flang-288613-3/Makefile
@@ -7,7 +7,7 @@ TESTSRC_ALL  = $(TESTSRC_MAIN) $(TESTSRC_AUX)
 
 FLANG        ?= flang
 OMP_BIN      = $(AOMP)/bin/$(FLANG)
-CC           = $(OMP_BIN) $(VERBOSE)
+CC           = $(OMP_BIN) $(VERBOSE) -mmlir --enable-gpu-heap-alloc
 #-ccc-print-phases
 #"-\#\#\#"
 
diff --git a/test/smoke/flang-288613/Makefile b/test/smoke/flang-288613/Makefile
index 2c1d032424..25f5c5f444 100644
--- a/test/smoke/flang-288613/Makefile
+++ b/test/smoke/flang-288613/Makefile
@@ -7,7 +7,7 @@ TESTSRC_ALL  = $(TESTSRC_MAIN) $(TESTSRC_AUX)
 
 FLANG        ?= flang
 OMP_BIN      = $(AOMP)/bin/$(FLANG)
-CC           = $(OMP_BIN) $(VERBOSE)
+CC           = $(OMP_BIN) $(VERBOSE) -mmlir --enable-gpu-heap-alloc
 #-ccc-print-phases
 #"-\#\#\#"
 
diff --git a/test/smoke/flang-288613/flang-288613.f90 b/test/smoke/flang-288613/flang-288613.f90
index bdd1f06705..dd020e7590 100644
--- a/test/smoke/flang-288613/flang-288613.f90
+++ b/test/smoke/flang-288613/flang-288613.f90
@@ -21,7 +21,7 @@ end subroutine dec_val_dev
         real(rstd) :: tmp
 #endif
 
-!$omp target teams
+!$omp target teams distribute
         do i=1,nsize
 !$omp parallel do simd private(tmp)
             do j=1,nsize
@@ -44,7 +44,7 @@ end subroutine dec_val_dev
                 end do
             end do
         end do
-!$omp end target teams
+!$omp end target teams distribute
     end subroutine _compute_dev
 
     subroutine compute_dev()