Merge pull request #1127 from pyiron/gpu_bug_fix

GPU bug fix
pyiron · Jun 9, 2023 · b717a0b · b717a0b
2 parents c034d0b + 455f99a
commit b717a0b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyiron_base/jobs/job/extension/server/generic.py b/pyiron_base/jobs/job/extension/server/generic.py
@@ -85,7 +85,7 @@ def __init__(
     ):
         self._cores = cores
         self._threads = threads
-        self._gpus = None
+        self._gpus = gpus
         self._run_time = None
         self._memory_limit = None
         self._host = self._init_host(host=host)
@@ -464,7 +464,7 @@ def to_hdf(self, hdf, group_name=None):
         if len(self.additional_arguments) > 0:
             hdf_dict["additional_arguments"] = self.additional_arguments
         if self._gpus is not None:
-            hdf_dict["accept_crash"] = self._gpus
+            hdf_dict["gpus"] = self._gpus
 
         if group_name is not None:
             with hdf.open(group_name) as hdf_group:
@@ -509,7 +509,7 @@ def from_hdf(self, hdf, group_name=None):
         if "additional_arguments" in hdf_dict.keys():
             self.additional_arguments = hdf_dict["additional_arguments"]
         if "gpus" in hdf_dict.keys():
-            self._gpus = hdf_dict["accept_crash"]
+            self._gpus = hdf_dict["gpus"]
         self._new_hdf = hdf_dict["new_h5"] == 1
 
     def db_entry(self):