SWDEV-532420 - Fix kokkos P2P copy failure with vmheap (#426)

Co-authored-by: Rahul Manocha <rmanocha@amd.com>
2025-07-07 17:27:13 -07:00
parent 9c1bff0ae7
commit 22b1ca4d8c
3 changed files with 13 additions and 11 deletions
@@ -246,7 +246,8 @@ address VmHeap::Alloc(size_t size) {
  } else {
    return nullptr;
  }
-  auto memory = new (base_memory_->getContext()) Buffer(*base_memory_, 0, offset, size);
+  auto memory =
+      new (device_->context()) Buffer(*base_memory_, 0, offset, size, &device_->context());
  if (nullptr == memory || !memory->create(nullptr)) {
    FreeBlock(hb);
    return nullptr;