From: Andrew Stubbs <ams@codesourcery.com>
Date: Mon, 6 Feb 2023 12:47:28 +0000 (+0000)
Subject: amdgcn: Pass -mstack-size through to runtime
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=196b25d7b9b73fa89ffb2f6dc73b851584f81127;p=thirdparty%2Fgcc.git

amdgcn: Pass -mstack-size through to runtime

But only for the offload case.

gcc/ChangeLog:

	* config/gcn/mkoffload.cc (gcn_stack_size): New global variable.
	(process_asm): Create a constructor for GCN_STACK_SIZE.
	(main): Parse the -mstack-size option.

(cherry picked from commit 45e01229af33a3dc3f124dcaec4b4ae11e9d07ce)
---

diff --git a/gcc/ChangeLog.omp b/gcc/ChangeLog.omp
index f7d0c439245f..4205a15ef7b8 100644
--- a/gcc/ChangeLog.omp
+++ b/gcc/ChangeLog.omp
@@ -1,3 +1,9 @@
+2023-02-06  Andrew Stubbs  <ams@codesourcery.com>
+
+	* config/gcn/mkoffload.cc (gcn_stack_size): New global variable.
+	(process_asm): Create a constructor for GCN_STACK_SIZE.
+	(main): Parse the -mstack-size option.
+
 2023-02-06  Andrew Stubbs  <ams@codesourcery.com>
 
 	* config/gcn/gcn-run.cc: Include libgomp-gcn.h.
diff --git a/gcc/config/gcn/mkoffload.cc b/gcc/config/gcn/mkoffload.cc
index 14d7eedc7907..3af3f717ea6c 100644
--- a/gcc/config/gcn/mkoffload.cc
+++ b/gcc/config/gcn/mkoffload.cc
@@ -123,6 +123,8 @@ uint32_t elf_arch = EF_AMDGPU_MACH_AMDGCN_GFX803;  // Default GPU architecture.
 uint32_t elf_flags =
     (EF_AMDGPU_FEATURE_XNACK_ANY_V4 | EF_AMDGPU_FEATURE_SRAMECC_ANY_V4);
 
+static int gcn_stack_size = 0;  /* Zero means use default.  */
+
 /* Delete tempfiles.  */
 
 void
@@ -701,6 +703,18 @@ process_asm (FILE *in, FILE *out, FILE *cfile)
 	      ? "val[0] != '1' || val[1] != '\\0'"
 	      : "val[0] == '1' && val[1] == '\\0'"));
 
+  /* Set the stack size if the user configured a value.  */
+  if (gcn_stack_size)
+    fprintf (cfile,
+	     "static __attribute__((constructor))\n"
+	     "void configure_stack_size (void)\n"
+	     "{\n"
+	     "  const char *val = getenv (\"GCN_STACK_SIZE\");\n"
+	     "  if (!val || val[0] == '\\0')\n"
+	     "    setenv (\"GCN_STACK_SIZE\", \"%d\", true);\n"
+	     "}\n\n",
+	     gcn_stack_size);
+
   obstack_free (&fns_os, NULL);
   for (i = 0; i < dims_count; i++)
     free (dims[i].name);
@@ -961,6 +975,10 @@ main (int argc, char **argv)
 	elf_arch = EF_AMDGPU_MACH_AMDGCN_GFX908;
       else if (strcmp (argv[i], "-march=gfx90a") == 0)
 	elf_arch = EF_AMDGPU_MACH_AMDGCN_GFX90a;
+#define STR "-mstack-size="
+      else if (startswith (argv[i], STR))
+	gcn_stack_size = atoi (argv[i] + strlen (STR));
+#undef STR
     }
 
   if (!(fopenacc ^ fopenmp))