SWDEV-514950 - fix hipMallocManaged & hipMemAdvise when capturing stream (#413)

lttamd · web-flow · commit d96016bcc5e7 · 2025-05-28T13:48:48.000+05:30
diff --git a/hipamd/src/hip_hmm.cpp b/hipamd/src/hip_hmm.cpp
@@ -72,6 +72,10 @@ hipError_t hipMallocManaged(void** dev_ptr, size_t size, unsigned int flags) {
     HIP_RETURN(hipErrorInvalidValue);
   }
 
+  if (!hip::tls.capture_streams_.empty() || !g_captureStreams.empty()) {
+    HIP_RETURN(hipErrorStreamCaptureUnsupported);
+  }
+
   HIP_RETURN(ihipMallocManaged(dev_ptr, size), *dev_ptr);
 }
 
@@ -151,6 +155,10 @@ hipError_t hipMemAdvise(const void* dev_ptr, size_t count, hipMemoryAdvise advic
     HIP_RETURN(hipErrorInvalidValue);
   }
 
+  if (!hip::tls.capture_streams_.empty() || !g_captureStreams.empty()) {
+    HIP_RETURN(hipErrorStreamCaptureUnsupported);
+  }
+
   size_t offset = 0;
   amd::Memory* memObj = getMemoryObject(dev_ptr, offset);
   if (memObj && count > (memObj->getSize() - offset)) {

Original file line number	Diff line number	Diff line change
`@@ -72,6 +72,10 @@ hipError_t hipMallocManaged(void** dev_ptr, size_t size, unsigned int flags) {`
`72`	`72`	`HIP_RETURN(hipErrorInvalidValue);`
`73`	`73`	`}`
`74`	`74`
	`75`	`+ if (!hip::tls.capture_streams_.empty() \|\| !g_captureStreams.empty()) {`
	`76`	`+ HIP_RETURN(hipErrorStreamCaptureUnsupported);`
	`77`	`+ }`
	`78`	`+`
`75`	`79`	`HIP_RETURN(ihipMallocManaged(dev_ptr, size), *dev_ptr);`
`76`	`80`	`}`
`77`	`81`
`@@ -151,6 +155,10 @@ hipError_t hipMemAdvise(const void* dev_ptr, size_t count, hipMemoryAdvise advic`
`151`	`155`	`HIP_RETURN(hipErrorInvalidValue);`
`152`	`156`	`}`
`153`	`157`
	`158`	`+ if (!hip::tls.capture_streams_.empty() \|\| !g_captureStreams.empty()) {`
	`159`	`+ HIP_RETURN(hipErrorStreamCaptureUnsupported);`
	`160`	`+ }`
	`161`	`+`
`154`	`162`	`size_t offset = 0;`
`155`	`163`	`amd::Memory* memObj = getMemoryObject(dev_ptr, offset);`
`156`	`164`	`if (memObj && count > (memObj->getSize() - offset)) {`