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
このコミットが含まれているのは:
Satyanvesh Dittakavi committed 2021-08-27 05:12:07 -04:00
1 親 e43e773283
コミット 169cc857fd
2ファイル変更
+8 -5

検索条件に一致するファイルはありません

+1 -1
ファイルの表示
@@ -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(