parent
a6de3b5d6f
commit
fc222d3325
|
@ -109,7 +109,7 @@ namespace nd4j {
|
||||||
|
|
||||||
if (deviceId != previousDeviceId) {
|
if (deviceId != previousDeviceId) {
|
||||||
// discard existing stuff
|
// discard existing stuff
|
||||||
nd4j_printf("AffinityManager::setCurrentDevice() was invoked, releasing buffers\n", "");
|
//nd4j_printf("AffinityManager::setCurrentDevice() was invoked, releasing buffers\n", "");
|
||||||
LaunchContext::releaseBuffers();
|
LaunchContext::releaseBuffers();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -158,7 +158,7 @@ LaunchContext::LaunchContext() {
|
||||||
};
|
};
|
||||||
|
|
||||||
void LaunchContext::releaseBuffers() {
|
void LaunchContext::releaseBuffers() {
|
||||||
nd4j_printf("LaunchContext::releaseBuffers() was invoked\n", "");
|
//nd4j_printf("LaunchContext::releaseBuffers() was invoked\n", "");
|
||||||
contextBuffers.release();
|
contextBuffers.release();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue