Blocking and default streams' sync:

Add hip::syncStreams(dev) to sync blocking streams on a given device.
hip::syncStreams(void) should only sync streams on the current device.

Change-Id: Ib6b0735215fa0ed12c646ebd029e9763ee3712ce
Этот коммит содержится в:
Christophe Paquot
2020-02-26 08:41:18 -08:00
родитель f4401ff2b5
Коммит fa6ed89cff
3 изменённых файлов: 24 добавлений и 11 удалений
+8 -2
Просмотреть файл
@@ -42,14 +42,20 @@ class StreamCallback {
namespace hip {
void syncStreams() {
void syncStreams(int devId) {
amd::ScopedLock lock(streamSetLock);
for (const auto& it : streamSet) {
it->finish();
if (it->device->deviceId() == devId) {
it->finish();
}
}
}
void syncStreams() {
syncStreams(getCurrentDevice()->deviceId());
}
Stream::Stream(hip::Device* dev, amd::CommandQueue::Priority p, unsigned int f) :
queue(nullptr), device(dev), priority(p), flags(f) {}