SWDEV-462192 SWDEV-459056 - Fixes corruption

SPT is destroyed with hipDeviceReset(). If a
stream is created right after reset, the same
object id could be reused. Later SPT destructor
incorrectly verifies that the stream is valid
referring to the reused object id causing the
corruption.

Change-Id: I3b1f7ffdf8bab874dca7b8fde22318162997b8f6


[ROCm/clr commit: f6a68b3c2e]
This commit is contained in:
Sourabh Betigeri authored and Sourabh Betigeri committed 2024-08-21 11:33:44 -04:00
1 parent b5acdd6fdc
commit d1d6c448c9
3 files changed
+6

No files matched your search

+1
View File
@@ -249,6 +249,7 @@ void Device::destroyAllStreams() {
for (auto& it : toBeDeleted) {
hip::Stream::Destroy(it);
}
hip::tls.stream_per_thread_obj_.clear_spt();
}
// ================================================================================================
+1
View File
@@ -269,6 +269,7 @@ public:
void operator=(const stream_per_thread& ) = delete;
~stream_per_thread();
hipStream_t get();
void clear_spt();
};
class Device;
+4
View File
@@ -236,6 +236,10 @@ hipStream_t stream_per_thread::get() {
return m_streams[currDev];
}
void stream_per_thread::clear_spt() {
m_streams[getCurrentDevice()->deviceId()] = nullptr;
}
// ================================================================================================
void getStreamPerThread(hipStream_t& stream) {