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