Maintain HIP_VISIBLE_DEVICES for kernel launch

Dieser Commit ist enthalten in:
Rahul Garg
2019-05-07 05:09:02 +05:30
Ursprung 117bdd8774
Commit 620a07102d
+4 -2
Datei anzeigen
@@ -2504,14 +2504,16 @@ hipError_t hipHccGetAcceleratorView(hipStream_t stream, hc::accelerator_view** a
namespace hip_impl { namespace hip_impl {
std::vector<hsa_agent_t> all_hsa_agents() { std::vector<hsa_agent_t> all_hsa_agents() {
std::vector<hsa_agent_t> r{}; 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(); const auto agent = acc.get_hsa_agent();
if (!agent || !acc.is_hsa_accelerator()) continue; if (!agent || !acc.is_hsa_accelerator()) continue;
r.emplace_back(*static_cast<hsa_agent_t*>(agent)); r.emplace_back(*static_cast<hsa_agent_t*>(agent));
} }
return r; return r;
} }