From 961440b59358d84f85b93e6f7d4e5ba1491a9025 Mon Sep 17 00:00:00 2001
From: Aditya Atluri <Aditya.Atluri@amd.com>
Date: Fri, 21 Jul 2017 15:50:12 -0500
Subject: [PATCH] fixed device selection during compilation to use
 rocm_agent_enumerator

1. Changed hipcc to use rocm_agent_enumerator
2. Changed square sample test to use device variable
---
 bin/hipcc                                | 22 ++++++++++++++++++++++
 samples/0_Intro/square/square.hipref.cpp |  5 +++--
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/bin/hipcc b/bin/hipcc
index 4a12b9e541..8a1874a36a 100755
--- a/bin/hipcc
+++ b/bin/hipcc
@@ -103,6 +103,28 @@ if ($HIP_PLATFORM eq "hcc") {
 
     $HIPLDFLAGS = `${HCC_HOME}/bin/hcc-config --ldflags`;
 
+    $ROCM_AGENT_ENUM = "${ROCM_PATH}/bin/rocm_agent_enumerator";
+
+    my $myAgents = `${ROCM_AGENT_ENUM} -t GPU`;
+    my @agentsLine = split('\n', $myAgents);
+
+    foreach my $val (@agentsLine) {
+        if($val eq "gfx701") {
+            $target_gfx701 = 1;
+        }
+        if($val eq "gfx801") {
+            $target_gfx801 = 1;
+        }
+        if($val eq "gfx802") {
+            $target_gfx802 = 1;
+        }
+        if($val eq "gfx803") {
+            $target_gfx803 = 1;
+        }
+        if($val eq "gfx900") {
+            $target_gfx900 = 1;
+        }
+    }
 
     #### GCC system includes workaround ####
     $HCC_WA_FLAGS = " ";
diff --git a/samples/0_Intro/square/square.hipref.cpp b/samples/0_Intro/square/square.hipref.cpp
index e694bfb8a4..167cb135f4 100644
--- a/samples/0_Intro/square/square.hipref.cpp
+++ b/samples/0_Intro/square/square.hipref.cpp
@@ -54,9 +54,10 @@ int main(int argc, char *argv[])
     float *A_h, *C_h;
     size_t N = 1000000;
     size_t Nbytes = N * sizeof(float);
-
+    static int device = 0;
+    CHECK(hipSetDevice(device));
     hipDeviceProp_t props;
-    CHECK(hipGetDeviceProperties(&props, 0/*deviceID*/));
+    CHECK(hipGetDeviceProperties(&props, device/*deviceID*/));
     printf ("info: running on device %s\n", props.name);
     #ifdef __HIP_PLATFORM_HCC__
       printf ("info: architecture on AMD GPU device is: %d\n",props.gcnArch);