feat: Enable install_gpu_drivers flag in v1 proto

Google APIs · copybara-github · commit b037bb3b4ed9 · 2022-10-14T12:56:13.000-07:00
docs: Refine GPU drivers installation proto description
docs: Refine comments for deprecated proto fields
docs: Update the API comments about the device_name

PiperOrigin-RevId: 481212993
diff --git a/google/cloud/batch/v1/job.proto b/google/cloud/batch/v1/job.proto
@@ -274,7 +274,9 @@ message AllocationPolicy {
     }
 
     // Device name that the guest operating system will see.
-    // If not specified, this is default to the disk name.
+    // It is used by Runnable.volumes field to mount disks. So please specify
+    // the device_name if you want Batch to help mount the disk, and it should
+    // match the device_name field in volumes.
     string device_name = 3;
   }
 
@@ -287,6 +289,7 @@ message AllocationPolicy {
     // The number of accelerators of this type.
     int64 count = 2;
 
+    // Deprecated: please use instances[0].install_gpu_drivers instead.
     bool install_gpu_drivers = 3 [deprecated = true];
   }
 
@@ -325,6 +328,12 @@ message AllocationPolicy {
       // c++ keyword conflict.
       string instance_template = 2;
     }
+
+    // Set this field true if users want Batch to help fetch drivers from a
+    // third party location and install them for GPUs specified in
+    // policy.accelerators or instance_template on their behalf. Default is
+    // false.
+    bool install_gpu_drivers = 3;
   }
 
   // A network interface.
@@ -378,6 +387,9 @@ message AllocationPolicy {
   // Only instances[0] is supported now.
   repeated InstancePolicyOrTemplate instances = 8;
 
+  // Service account that VMs will run as.
+  ServiceAccount service_account = 9;
+
   // Labels applied to all VM instances and other resources
   // created by AllocationPolicy.
   // Labels could be user provided or system generated.
@@ -443,3 +455,12 @@ message TaskGroup {
   // VMs running the Batch tasks in the same TaskGroup.
   bool permissive_ssh = 12;
 }
+
+// Carries information about a Google Cloud service account.
+message ServiceAccount {
+  // Email address of the service account. If not specified, the default
+  // Compute Engine service account for the project will be used. If instance
+  // template is being used, the service account has to be specified in the
+  // instance template and it has to match the email field here.
+  string email = 1;
+}
