Add GPU stats to allocation wrapper API. (#1016)

AcademySoftwareFoundation · Sep 12, 2021 · 341a6d1 · 341a6d1
1 parent e743be1
commit 341a6d1
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/pycue/opencue/wrappers/allocation.py b/pycue/opencue/wrappers/allocation.py
@@ -167,6 +167,46 @@ def totalLockedCores(self):
  """
  return self.data.stats.locked_cores
 
+ def totalGpus(self):
+ """Returns the total number of gpus in the allocation.
+
+ :rtype: float
+ :return: total number of gpus in the allocation
+ """
+ return self.data.stats.gpus
+
+ def totalAvailableGpus(self):
+ """Returns the total number of gpus available for booking in the allocation.
+
+ :rtype: float
+ :return: total number of gpus in the allocation
+ """
+ return self.data.stats.available_gpus
+
+ def totalIdleGpus(self):
+ """Returns the total number of idle gpus in the allocation.
+
+ :rtype: float
+ :return: total number of idle gpus in the allocation
+ """
+ return self.data.stats.idle_gpus
+
+ def totalRunningGpus(self):
+ """Returns the total number of running gpus in the allocation.
+
+ :rtype: float
+ :return: total number of running gpus in the allocation
+ """
+ return self.data.stats.running_gpus
+
+ def totalLockedGpus(self):
+ """Returns the total number of locked gpus in the allocation.
+
+ :rtype: float
+ :return: total number of locked gpus in the allocation
+ """
+ return self.data.stats.locked_gpus
+
  def totalHosts(self):
  """Returns the total number of hosts in the allocation.