Added isManaged to if statement check

oneapi-src · Mar 5, 2024 · e6dfc62 · e6dfc62
1 parent 19df225
commit e6dfc62
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/source/adapters/hip/enqueue.cpp b/source/adapters/hip/enqueue.cpp
@@ -807,7 +807,10 @@ ur_result_t commonMemSetLargePattern(hipStream_t Stream, uint32_t PatternSize,
  hipPointerAttribute_t ptrAttribs{};
  UR_CHECK_ERROR(hipPointerGetAttributes(&ptrAttribs, (const void *)Ptr));
 
- if (ptrAttribs.hostPointer) {
+ // The hostPointer attribute is non-null also for shared memory allocations.
+ // To make sure that this workaround only executes for host pinned memory, we
+ // need to check that isManaged attribute is false.
+ if (ptrAttribs.hostPointer && !ptrAttribs.isManaged) {
  const auto NumOfCopySteps = Size / PatternSize;
  const auto Offset = sizeof(uint32_t);
  const auto LeftPatternSize = PatternSize - Offset;