Skip to content

Commit 3781749

Browse files
authored
fix compilation error (#5061)
1 parent cada33d commit 3781749

File tree

4 files changed

+15
-5
lines changed

4 files changed

+15
-5
lines changed

source/module_hamilt_lcao/module_gint/gint_force_gpu.cu

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,11 @@ void gint_fvl_gpu(const hamilt::HContainer<double>* dm,
3131
const Grid_Technique& gridt,
3232
const UnitCell& ucell)
3333
{
34-
int dev_id = base_device::information::set_device_by_rank();
34+
#ifdef __MPI
35+
const int dev_id = base_device::information::set_device_by_rank();
36+
#else
37+
const int dev_id = 0;
38+
#endif
3539
// checkCuda(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync));
3640

3741
const int nbzp = gridt.nbzp;

source/module_hamilt_lcao/module_gint/gint_rho_gpu.cu

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,11 @@ void gint_rho_gpu(const hamilt::HContainer<double>* dm,
1717
const UnitCell& ucell,
1818
double* rho)
1919
{
20-
int dev_id = base_device::information::set_device_by_rank();
20+
#ifdef __MPI
21+
const int dev_id = base_device::information::set_device_by_rank();
22+
#else
23+
const int dev_id = 0;
24+
#endif
2125
// checkCuda(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync));
2226

2327
const int nbzp = gridt.nbzp;

source/module_hamilt_lcao/module_gint/gint_vl_gpu.cu

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,11 @@ void gint_vl_gpu(hamilt::HContainer<double>* hRGint,
2929
double* pvpR,
3030
const bool is_gamma_only)
3131
{
32-
int dev_id = base_device::information::set_device_by_rank();
32+
#ifdef __MPI
33+
const int dev_id = base_device::information::set_device_by_rank();
34+
#else
35+
const int dev_id = 0;
36+
#endif
3337
// checkCuda(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync));
3438
const int nbzp = gridt.nbzp;
3539
const int num_streams = gridt.nstreams;

source/module_hamilt_lcao/module_gint/grid_technique.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -560,8 +560,6 @@ int Grid_Technique::find_offset(const int id1, const int id2, const int iat1, co
560560

561561
void Grid_Technique::init_gpu_gint_variables(const UnitCell& ucell,
562562
const int num_stream) {
563-
564-
int dev_id = base_device::information::set_device_by_rank();
565563
if (is_malloced) {
566564
free_gpu_gint_variables(this->nat);
567565
}

0 commit comments

Comments
 (0)