SWDEV-298985 - hipMemPrefetchAsync should prefetch the data to the specified destination device

Pass the device agent specified by the user to the ROCr api instead of passing the device agent attached to the specified stream

Change-Id: I86c98935b9dc404eaa6d47ccdd082a8c3678fb36
Este cometimento está contido em:
Satyanvesh Dittakavi cometeu 2021-08-27 05:12:07 -04:00
1 ascendente e43e773283
cometimento 169cc857fd
2 ficheiros modificados
+8 -5

Não existem ficheiros que correspondam à sua pesquisa

+1 -1
Ver ficheiro
@@ -1578,7 +1578,7 @@ void VirtualGPU::submitSvmPrefetchAsync(amd::SvmPrefetchAsyncCommand& cmd) {
// Find the requested agent for the transfer
hsa_agent_t agent = (cmd.cpu_access() ||
(dev().settings().hmmFlags_ & Settings::Hmm::EnableSystemMemory)) ?
dev().getCpuAgent() : gpu_device();
dev().getCpuAgent() : (static_cast<const roc::Device*>(cmd.device()))->getBackendDevice();
// Initiate a prefetch command
hsa_status_t status = hsa_amd_svm_prefetch_async(