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

Co-authored-by: Rahul Manocha <rmanocha@amd.com>

[ROCm/clr commit: 22b1ca4d8c]

projects/clr/rocclr/platform/vmheap.cpp

@@ -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;