AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-381402 - Derive hip::Stream from amd::HostQueue

Browse Source 
Change-Id: I6c1aca5eb350c32d974ae4ffcc725705355956d8
This commit is contained in:
Ioannis Assiouras
2023-02-08 20:18:11 +00:00
parent aaf1c19609
commit e3633dc8f4
19 changed files with 403 additions and 732 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_device.cpp
+20 -10
View File
@@ -26,25 +26,31 @@
namespace hip {
// ================================================================================================
amd::HostQueue* Device::NullStream(bool skip_alloc) {
amd::HostQueue* null_queue = null_stream_.asHostQueue(skip_alloc);
if (null_queue == nullptr) {
hip::Stream* Device::NullStream(bool skip_alloc) {
if (null_stream_ == nullptr && !skip_alloc) {
null_stream_ = new Stream(this, Stream::Priority::Normal, 0, true);
}
if (null_stream_ == nullptr) {
return nullptr;
}
// Wait for all active streams before executing commands on the default
iHipWaitActiveStreams(null_queue);
return null_queue;
iHipWaitActiveStreams(null_stream_);
return null_stream_;
}
// ================================================================================================
Stream* Device::GetNullStream() {
amd::HostQueue* null_queue = null_stream_.asHostQueue();
if (null_queue == nullptr) {
hip::Stream* Device::GetNullStream() {
if (null_stream_ == nullptr) {
null_stream_ = new Stream(this, Stream::Priority::Normal, 0, true);
}
if (null_stream_ == nullptr) {
return nullptr;
}
// Wait for all active streams before executing commands on the default
iHipWaitActiveStreams(null_queue);
return &null_stream_;
iHipWaitActiveStreams(null_stream_);
return null_stream_;
}
// ================================================================================================
@@ -128,6 +134,10 @@ Device::~Device() {
if (default_mem_pool_ != nullptr) {
default_mem_pool_->release();
}
if (null_stream_!= nullptr) {
delete null_stream_;
}
}
}