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(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))) {