diff options
Diffstat (limited to 'src/core/cpu/device.cpp')
-rw-r--r-- | src/core/cpu/device.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/core/cpu/device.cpp b/src/core/cpu/device.cpp index a8d6aaf..254bcbd 100644 --- a/src/core/cpu/device.cpp +++ b/src/core/cpu/device.cpp @@ -372,9 +372,11 @@ cl_int CPUDevice::createSubDevices( if (partition_size) { // Create num_new_devices SubDevices: Coal::CPUDevice * new_device; - for (int i = 0; i < num_new_devices; i++) { - new_device = new CPUDevice(this, partition_size); - if (out_devices) out_devices[i] = (cl_device_id)new_device; + if (out_devices) { + for (int i = 0; i < num_new_devices; i++) { + new_device = new CPUDevice(this, partition_size); + out_devices[i] = (cl_device_id)new_device; + } } if (num_devices_ret) *num_devices_ret = num_new_devices; } |