fix a minor bug on reporting the selected device.
This commit is contained in:
1 parent
48f6f9c25e
commit
ee33aacdd6
1 file changed
+1
-1
@@ -245,7 +245,7 @@ int CudaBase::cuda_setDevice(int device) {
|
||||
std::cout << "Init: " << device << "\t" << ndev << std::endl;
|
||||
|
||||
if (device < ndev) {
|
||||
std::cout << "set device to: " << ndev << std::endl;
|
||||
std::cout << "set device to: " << device << std::endl;
|
||||
cudaSetDevice(device);
|
||||
} else {
|
||||
if (ndev > 0)
|
||||
|
||||
Reference in new issue
Block a user