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;
|
std::cout << "Init: " << device << "\t" << ndev << std::endl;
|
||||||
|
|
||||||
if (device < ndev) {
|
if (device < ndev) {
|
||||||
std::cout << "set device to: " << ndev << std::endl;
|
std::cout << "set device to: " << device << std::endl;
|
||||||
cudaSetDevice(device);
|
cudaSetDevice(device);
|
||||||
} else {
|
} else {
|
||||||
if (ndev > 0)
|
if (ndev > 0)
|
||||||
|
Reference in New Issue
Block a user