Skip to content

Commit b5815c6

Browse files
committed
simplify set device
1 parent 7187180 commit b5815c6

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

source/module_hsolver/kernels/cuda/diag_cusolver.cu

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,7 @@
66
Diag_Cusolver_gvd::Diag_Cusolver_gvd(){
77
// step 1: create cusolver/cublas handle
88
#if defined(__MPI) && defined(__CUDA)
9-
device_id = base_device::information::set_device_by_rank();
10-
checkCudaErrors( cudaSetDevice(device_id) );
9+
base_device::information::set_device_by_rank();
1110
#endif
1211
cusolverH = NULL;
1312
checkCudaErrors( cusolverDnCreate(&cusolverH) );

source/module_hsolver/kernels/cuda/diag_cusolver.cuh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ class Diag_Cusolver_gvd{
2222
//-------------------
2323

2424
cusolverDnHandle_t cusolverH = nullptr;
25-
int device_id = 0;
2625

2726
cusolverEigType_t itype = CUSOLVER_EIG_TYPE_1; //problem type: A*x = (lambda)*B*x
2827
cusolverEigMode_t jobz = CUSOLVER_EIG_MODE_NOVECTOR; // compute eigenvalues and eigenvectors.

0 commit comments

Comments
 (0)