From 3f65bec096ab1d908c69f9f29db981e93f8a709a Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Tue, 7 May 2019 05:09:02 +0530
Subject: [PATCH] Maintain HIP_VISIBLE_DEVICES for kernel launch

[ROCm/clr commit: 3be54a903c1ff6af744beca1d8fed782cf55ff65]
---
 projects/clr/hipamd/src/hip_hcc.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/projects/clr/hipamd/src/hip_hcc.cpp b/projects/clr/hipamd/src/hip_hcc.cpp
index ac285e6a83..6f2e41cd49 100644
--- a/projects/clr/hipamd/src/hip_hcc.cpp
+++ b/projects/clr/hipamd/src/hip_hcc.cpp
@@ -2504,14 +2504,16 @@ hipError_t hipHccGetAcceleratorView(hipStream_t stream, hc::accelerator_view** a
 namespace hip_impl {
     std::vector<hsa_agent_t> all_hsa_agents() {
         std::vector<hsa_agent_t> r{};
-        for (auto&& acc : hc::accelerator::get_all()) {
+        std::vector<hc::accelerator> visible_accelerators;
+        for (int i=0; i < g_deviceCnt; i++)
+            visible_accelerators.push_back(g_deviceArray[i]->_acc);
+        for (auto&& acc : visible_accelerators) {
             const auto agent = acc.get_hsa_agent();
 
             if (!agent || !acc.is_hsa_accelerator()) continue;
 
             r.emplace_back(*static_cast<hsa_agent_t*>(agent));
         }
-
         return r;
     }