From 014f1797d9dce095fc6a76f8bb920a94f0bb51de Mon Sep 17 00:00:00 2001
From: sunway513 <peng.sun@amd.com>
Date: Thu, 27 Jul 2017 17:41:49 +0000
Subject: [PATCH] Change to use hipDeviceGetPCIBusId api for hipEnvVarDriver
 test

[ROCm/hip commit: ed0d6cf7454d792d3d2edd7ac72dcee07b5dc974]
---
 projects/hip/tests/src/hipEnvVar.cpp       | 15 ++++----
 projects/hip/tests/src/hipEnvVarDriver.cpp | 42 +++++++++++-----------
 2 files changed, 27 insertions(+), 30 deletions(-)

diff --git a/projects/hip/tests/src/hipEnvVar.cpp b/projects/hip/tests/src/hipEnvVar.cpp
index d5942518e0..ff68103816 100644
--- a/projects/hip/tests/src/hipEnvVar.cpp
+++ b/projects/hip/tests/src/hipEnvVar.cpp
@@ -109,15 +109,14 @@ int main(int argc, char **argv)
         std::cout << devCount  << std::endl;
     }
     if (retDevInfo) {
-        hipSetDevice(device);
-        hipDeviceProp_t devProp;
+	hipDevice_t deviceT;
+	hipDeviceGet(&deviceT, device);
 
-        hipGetDeviceProperties(&devProp, device);
-        if (devProp.major < 1) {
-            printf("%d does not support HIP\n", device);
-            return -1;
-        }
-        std::cout << devProp.pciBusID << std::endl;
+        char pciBusId[100];
+        memset(pciBusId,0,100);
+        hipDeviceGetPCIBusId(pciBusId,100,deviceT);
+
+        cout<<pciBusId<<endl;
     }
     exit(0);
 }
diff --git a/projects/hip/tests/src/hipEnvVarDriver.cpp b/projects/hip/tests/src/hipEnvVarDriver.cpp
index dc1d451e0d..c3b914ece9 100644
--- a/projects/hip/tests/src/hipEnvVarDriver.cpp
+++ b/projects/hip/tests/src/hipEnvVarDriver.cpp
@@ -40,7 +40,7 @@ int getDeviceNumber(){
     if(!(in = popen("./directed_tests/hipEnvVar -c", "r"))){
          return 1;
      }
-    while(fgets(buff, sizeof(buff), in)!=NULL){
+    while(fgets(buff, 512, in)!=NULL){
          cout << buff;
     }
     pclose(in);
@@ -48,47 +48,44 @@ int getDeviceNumber(){
 }
 
 // Query the current device ID remotely to hipEnvVar
-int getDevicePCIBusNumRemote(int deviceID){
+void getDevicePCIBusNumRemote(int deviceID, char* pciBusID){
     FILE *in;
-    char buff[512];
     string str = "./directed_tests/hipEnvVar -d ";
     str += std::to_string(deviceID);
     std::this_thread::sleep_for(std::chrono::milliseconds(10));
     if(!(in = popen(str.c_str(), "r"))){
-        return 1;
+        exit(1);
     }
-    while(fgets(buff, sizeof(buff), in)!=NULL){
-        cout << buff;
+    while(fgets(pciBusID, 100, in)!=NULL){
+        cout << pciBusID;
     }
     pclose(in);
-    return atoi(buff);
 }
 
-// Query the current device ID locally
-int getDevicePCIBusNum(int deviceID){
-    hipSetDevice(deviceID);
-    hipDeviceProp_t devProp;
+// Query the current device ID locally on AMD path
+void getDevicePCIBusNum(int deviceID, char* pciBusID){
+    hipDevice_t deviceT;
+    hipDeviceGet(&deviceT, deviceID);
 
-    hipGetDeviceProperties(&devProp, deviceID);
-    if (devProp.major < 1) {
-        printf("%d does not support HIP\n", deviceID);
-        return -1;
-    }
-    return devProp.pciBusID;
+    memset(pciBusID,0,100);
+    hipDeviceGetPCIBusId(pciBusID,100,deviceT);
 }
 
 int main() {
     unsetenv("HIP_VISIBLE_DEVICES");
     unsetenv("CUDA_VISIBLE_DEVICES");
+    
+    std::vector<std::string> devPCINum;
+    char pciBusID[100]; 
     //collect the device pci bus ID for all devices
     int totalDeviceNum = getDeviceNumber();
     std::cout << "The total number of available devices is " << totalDeviceNum<< std::endl
         <<"Valid index range is 0 - "<<totalDeviceNum-1<<std::endl;
-    std::vector<int> devPCINum;
     for (int i = 0; i < totalDeviceNum ; i++) {
-        devPCINum.push_back(getDevicePCIBusNum(i));
+        getDevicePCIBusNum(i, pciBusID);
+        devPCINum.push_back(pciBusID);
         std::cout <<"The collected device PCI Bus ID of Device "<<i<<" is "
-            << getDevicePCIBusNum(i) << std::endl;
+            << devPCINum.back() << std::endl;
     }
 
     //select each of the available devices to be the target device,
@@ -96,9 +93,10 @@ int main() {
     for (int i = 0; i < totalDeviceNum ; i++) {
         setenv("HIP_VISIBLE_DEVICES",(char*)std::to_string(i).c_str(),1);
         setenv("CUDA_VISIBLE_DEVICES",(char*)std::to_string(i).c_str(),1);
-        if (devPCINum[i] != getDevicePCIBusNumRemote(0)) {
+	getDevicePCIBusNumRemote(0, pciBusID);
+        if (devPCINum[i] == pciBusID) {
             std::cout << "The returned PciBusID is not correct"<< std::endl;
-            std::cout << "Expected "<< devPCINum[i] << ", but get " << getDevicePCIBusNum(i) << endl;
+            std::cout << "Expected "<< devPCINum[i] << ", but get " << pciBusID << endl;
             exit(-1);
         } else {
             continue;