fix a minor bug on reporting the selected device.
This commit is contained in:
@ -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