From dda2613525512c4a558daa4773484541b75d77ca Mon Sep 17 00:00:00 2001
From: Satyanvesh Dittakavi <Satyanvesh.Dittakavi@amd.com>
Date: Mon, 30 Aug 2021 13:36:35 +0000
Subject: [PATCH] SWDEV-300934 - Check if the stream is valid

Change-Id: I41797dc3e726c7eea6a405f20deebd6f59381b5e
---
 hipamd/src/hip_hmm.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/hipamd/src/hip_hmm.cpp b/hipamd/src/hip_hmm.cpp
index fd18b2b6f9..bedc2f1029 100644
--- a/hipamd/src/hip_hmm.cpp
+++ b/hipamd/src/hip_hmm.cpp
@@ -80,6 +80,10 @@ hipError_t hipMemPrefetchAsync(const void* dev_ptr, size_t count, int device,
     HIP_RETURN(hipErrorInvalidValue);
   }
 
+  if (!hip::isValid(stream)) {
+    HIP_RETURN(hipErrorContextIsDestroyed);
+  }
+
   size_t offset = 0;
   amd::Memory* memObj = getMemoryObject(dev_ptr, offset);
   // Return error if count passed is more than the actual size allocated