GPU-limits proto message (#227)

rigdev · Oct 24, 2023 · c9dde0f · c9dde0f
1 parent f75bade
commit c9dde0f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/proto/rig/api/v1/capsule/rollout.proto b/proto/rig/api/v1/capsule/rollout.proto
@@ -88,6 +88,7 @@ message ContainerSettings {
 message Resources {
   ResourceList requests = 1;
   ResourceList limits = 2;
+  GpuLimits gpu_limits = 3;
 }
 
 message ResourceList {
@@ -97,6 +98,15 @@ message ResourceList {
   uint64 memory_bytes = 2;
 }
 
+message GpuLimits {
+  // gpu resource type - fx. nvidia.com/gpu
+  string type = 1;
+  // number of gpus
+  uint32 count = 2;
+}
+
+
+
 message Network {
   repeated Interface interfaces = 1;
 }