fix config

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Nov 13, 2024 · cfff834 · cfff834
1 parent 39f357c
commit cfff834
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/api_cc/src/DeepPotJAX.cc b/source/api_cc/src/DeepPotJAX.cc
@@ -249,9 +249,9 @@ void deepmd::DeepPotJAX::init(const std::string& model,
   double gpu_memory_fraction = 0.9;
   auto bytes = reinterpret_cast<std::uint8_t*>(&gpu_memory_fraction);
 
-  // Put it to the config byte-array, from 3 to 10:
+  // Put it to the config byte-array, from 7 to 14:
   for (std::size_t i = 0; i < sizeof(gpu_memory_fraction); ++i) {
-    config[i + 3] = bytes[i];
+    config[i + 7] = bytes[i];
   }
 
   TF_SetConfig(sessionopts, config.data(), config.size(), status);