From 6deecf1bfef512136dfc63540c40c7459745b320 Mon Sep 17 00:00:00 2001
From: Vladana Stojiljkovic <Vladana.Stojiljkovic@amd.com>
Date: Wed, 9 Oct 2024 16:49:27 +0200
Subject: [PATCH] SWDEV-490474 - Allow hipMallocManaged capturing only in
 relaxed mode

Change-Id: I02dccc6c45e39082ef925509a28bbe3c2a0fb7c6
---
 hipamd/src/hip_hmm.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hipamd/src/hip_hmm.cpp b/hipamd/src/hip_hmm.cpp
index 95c4c44a6c..23b759f481 100644
--- a/hipamd/src/hip_hmm.cpp
+++ b/hipamd/src/hip_hmm.cpp
@@ -64,6 +64,7 @@ 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))) {