From 69112331acd0edd10ae7524eb3d4495f13ae34d8 Mon Sep 17 00:00:00 2001
From: "Stojiljkovic, Vladana" <Vladana.Stojiljkovic@amd.com>
Date: Thu, 22 May 2025 17:21:30 +0200
Subject: [PATCH] Allow hipMallocManaged capturing only in relaxed mode (#383)

Co-authored-by: Anusha GodavarthySurya <Anusha.GodavarthySurya@amd.com>
---
 hipamd/src/hip_hmm.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/hipamd/src/hip_hmm.cpp b/hipamd/src/hip_hmm.cpp
index 5f636d6b39..64f2ca0435 100644
--- a/hipamd/src/hip_hmm.cpp
+++ b/hipamd/src/hip_hmm.cpp
@@ -65,6 +65,8 @@ static_assert(static_cast<uint32_t>(hipMemRangeAttributeLastPrefetchLocation) ==
 hipError_t hipMallocManaged(void** dev_ptr, size_t size, unsigned int flags) {
   HIP_INIT_API(hipMallocManaged, dev_ptr, size, flags);
 
+  CHECK_STREAM_CAPTURE_SUPPORTED();
+
   if ((dev_ptr == nullptr) || (size == 0) ||
       ((flags != hipMemAttachGlobal) && (flags != hipMemAttachHost))) {
     HIP_RETURN(hipErrorInvalidValue);