[llvm] [OpenMP][Offload] Update the Logic for Configuring Auto Zero-Copy (PR #143638)
via llvm-commits
llvm-commits at lists.llvm.org
Tue Jun 10 22:24:07 PDT 2025
================
@@ -286,16 +286,16 @@ void PluginManager::registerLib(__tgt_bin_desc *Desc) {
}
PM->RTLsMtx.unlock();
- bool UseAutoZeroCopy = Plugins.size() > 0;
+ bool UseAutoZeroCopy = false;
auto ExclusiveDevicesAccessor = getExclusiveDevicesAccessor();
- for (const auto &Device : *ExclusiveDevicesAccessor)
- UseAutoZeroCopy &= Device->useAutoZeroCopy();
+ // APUs are homogeneous set of GPUs. Check the first device for
+ // configuring Auto Zero-Copy.
+ if (ExclusiveDevicesAccessor->size() > 0) {
+ auto &Device = *(*ExclusiveDevicesAccessor)[0];
+ UseAutoZeroCopy = Device.useAutoZeroCopy();
----------------
Kewen12 wrote:
That's what mentioned in Carlo's reply. We'd be happy to revisit in the future if needed. Also, this change would work better for our future changes on device initialization.
https://github.com/llvm/llvm-project/pull/143638
More information about the llvm-commits
mailing list