amdgcn: Clean up device memory in gcn-run

gcc/ChangeLog:

	* config/gcn/gcn-run.cc (main): Add an hsa_memory_free calls for each
	device_malloc call.

diff --git a/gcc/config/gcn/gcn-run.cc b/gcc/config/gcn/gcn-run.cc
index d45ff3e6c2ba9722f80efc78e1a53d7ec3d4d38c..2f3ed2d41d2fc35a48e822f02bd4af4d5c4defb4 100644 (file)
--- a/gcc/config/gcn/gcn-run.cc
+++ b/gcc/config/gcn/gcn-run.cc
@@ -755,7 +755,13 @@ main (int argc, char *argv[])
 
   /* Clean shut down.  */
   XHSA (hsa_fns.hsa_memory_free_fn (kernargs),
-       "Clean up device memory");
+       "Clean up device kernargs memory");
+  XHSA (hsa_fns.hsa_memory_free_fn (args),
+       "Clean up device args memory");
+  XHSA (hsa_fns.hsa_memory_free_fn (heap),
+       "Clean up device heap memory");
+  XHSA (hsa_fns.hsa_memory_free_fn (stack),
+       "Clean up device stack memory");
   XHSA (hsa_fns.hsa_executable_destroy_fn (executable),
        "Clean up GCN executable");
   XHSA (hsa_fns.hsa_queue_destroy_fn (queue),