[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Mar 28, 2024 · 811ff3a · 811ff3a
1 parent de14a32
commit 811ff3a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/source/api_cc/src/DeepPotPT.cc b/source/api_cc/src/DeepPotPT.cc
@@ -38,10 +38,11 @@ void DeepPotPT::init(const std::string& model,
     return;
   }
   int gpu_num = torch::cuda::device_count();
-  if(gpu_num > 0)
+  if (gpu_num > 0) {
     gpu_id = gpu_rank % gpu_num;
-  else
+  } else {
     gpu_id = 0;
+  }
   torch::Device device(torch::kCUDA, gpu_id);
   gpu_enabled = torch::cuda::is_available();
   if (!gpu_enabled) {