Skip to content

Commit a99144f

Browse files
gcf-owl-bot[bot]amanda-tarafa
authored andcommitted
feat: Add multihost_gpu_node_count to Vertex SDK for multihost GPU support
PiperOrigin-RevId: 733768886 Source-Link: googleapis/googleapis@4129cd8 Source-Link: googleapis/googleapis-gen@fc4062c Copy-Tag: eyJwIjoiYXBpcy9Hb29nbGUuQ2xvdWQuQUlQbGF0Zm9ybS5WMUJldGExLy5Pd2xCb3QueWFtbCIsImgiOiJmYzQwNjJjY2MyOTczZDNkY2RjNWVhMjI0ZTc5ODEzMDdiNTNjMjE3In0=
1 parent 33f9df1 commit a99144f

File tree

1 file changed

+72
-30
lines changed

1 file changed

+72
-30
lines changed

apis/Google.Cloud.AIPlatform.V1Beta1/Google.Cloud.AIPlatform.V1Beta1/MachineResources.g.cs

Lines changed: 72 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -29,43 +29,44 @@ static MachineResourcesReflection() {
2929
"Gh9nb29nbGUvYXBpL2ZpZWxkX2JlaGF2aW9yLnByb3RvGjZnb29nbGUvY2xv",
3030
"dWQvYWlwbGF0Zm9ybS92MWJldGExL2FjY2VsZXJhdG9yX3R5cGUucHJvdG8a",
3131
"Omdvb2dsZS9jbG91ZC9haXBsYXRmb3JtL3YxYmV0YTEvcmVzZXJ2YXRpb25f",
32-
"YWZmaW5pdHkucHJvdG8iiwIKC01hY2hpbmVTcGVjEhkKDG1hY2hpbmVfdHlw",
32+
"YWZmaW5pdHkucHJvdG8itQIKC01hY2hpbmVTcGVjEhkKDG1hY2hpbmVfdHlw",
3333
"ZRgBIAEoCUID4EEFEk8KEGFjY2VsZXJhdG9yX3R5cGUYAiABKA4yMC5nb29n",
3434
"bGUuY2xvdWQuYWlwbGF0Zm9ybS52MWJldGExLkFjY2VsZXJhdG9yVHlwZUID",
3535
"4EEFEhkKEWFjY2VsZXJhdG9yX2NvdW50GAMgASgFEhkKDHRwdV90b3BvbG9n",
36-
"eRgEIAEoCUID4EEFEloKFHJlc2VydmF0aW9uX2FmZmluaXR5GAUgASgLMjQu",
37-
"Z29vZ2xlLmNsb3VkLmFpcGxhdGZvcm0udjFiZXRhMS5SZXNlcnZhdGlvbkFm",
38-
"ZmluaXR5QgbgQQXgQQEiugIKEkRlZGljYXRlZFJlc291cmNlcxJKCgxtYWNo",
36+
"eRgEIAEoCUID4EEFEigKGG11bHRpaG9zdF9ncHVfbm9kZV9jb3VudBgGIAEo",
37+
"BUIG4EEF4EEBEloKFHJlc2VydmF0aW9uX2FmZmluaXR5GAUgASgLMjQuZ29v",
38+
"Z2xlLmNsb3VkLmFpcGxhdGZvcm0udjFiZXRhMS5SZXNlcnZhdGlvbkFmZmlu",
39+
"aXR5QgbgQQXgQQEiugIKEkRlZGljYXRlZFJlc291cmNlcxJKCgxtYWNoaW5l",
40+
"X3NwZWMYASABKAsyLC5nb29nbGUuY2xvdWQuYWlwbGF0Zm9ybS52MWJldGEx",
41+
"Lk1hY2hpbmVTcGVjQgbgQQLgQQUSIQoRbWluX3JlcGxpY2FfY291bnQYAiAB",
42+
"KAVCBuBBAuBBBRIeChFtYXhfcmVwbGljYV9jb3VudBgDIAEoBUID4EEFEiMK",
43+
"FnJlcXVpcmVkX3JlcGxpY2FfY291bnQYCSABKAVCA+BBARJdChhhdXRvc2Nh",
44+
"bGluZ19tZXRyaWNfc3BlY3MYBCADKAsyNi5nb29nbGUuY2xvdWQuYWlwbGF0",
45+
"Zm9ybS52MWJldGExLkF1dG9zY2FsaW5nTWV0cmljU3BlY0ID4EEFEhEKBHNw",
46+
"b3QYBSABKAhCA+BBASJUChJBdXRvbWF0aWNSZXNvdXJjZXMSHgoRbWluX3Jl",
47+
"cGxpY2FfY291bnQYASABKAVCA+BBBRIeChFtYXhfcmVwbGljYV9jb3VudBgC",
48+
"IAEoBUID4EEFIqoBChdCYXRjaERlZGljYXRlZFJlc291cmNlcxJKCgxtYWNo",
3949
"aW5lX3NwZWMYASABKAsyLC5nb29nbGUuY2xvdWQuYWlwbGF0Zm9ybS52MWJl",
40-
"dGExLk1hY2hpbmVTcGVjQgbgQQLgQQUSIQoRbWluX3JlcGxpY2FfY291bnQY",
41-
"AiABKAVCBuBBAuBBBRIeChFtYXhfcmVwbGljYV9jb3VudBgDIAEoBUID4EEF",
42-
"EiMKFnJlcXVpcmVkX3JlcGxpY2FfY291bnQYCSABKAVCA+BBARJdChhhdXRv",
43-
"c2NhbGluZ19tZXRyaWNfc3BlY3MYBCADKAsyNi5nb29nbGUuY2xvdWQuYWlw",
44-
"bGF0Zm9ybS52MWJldGExLkF1dG9zY2FsaW5nTWV0cmljU3BlY0ID4EEFEhEK",
45-
"BHNwb3QYBSABKAhCA+BBASJUChJBdXRvbWF0aWNSZXNvdXJjZXMSHgoRbWlu",
46-
"X3JlcGxpY2FfY291bnQYASABKAVCA+BBBRIeChFtYXhfcmVwbGljYV9jb3Vu",
47-
"dBgCIAEoBUID4EEFIqoBChdCYXRjaERlZGljYXRlZFJlc291cmNlcxJKCgxt",
48-
"YWNoaW5lX3NwZWMYASABKAsyLC5nb29nbGUuY2xvdWQuYWlwbGF0Zm9ybS52",
49-
"MWJldGExLk1hY2hpbmVTcGVjQgbgQQLgQQUSIwoWc3RhcnRpbmdfcmVwbGlj",
50-
"YV9jb3VudBgCIAEoBUID4EEFEh4KEW1heF9yZXBsaWNhX2NvdW50GAMgASgF",
51-
"QgPgQQUiLwoRUmVzb3VyY2VzQ29uc3VtZWQSGgoNcmVwbGljYV9ob3VycxgB",
52-
"IAEoAUID4EEDIj0KCERpc2tTcGVjEhYKDmJvb3RfZGlza190eXBlGAEgASgJ",
53-
"EhkKEWJvb3RfZGlza19zaXplX2diGAIgASgFIj0KElBlcnNpc3RlbnREaXNr",
54-
"U3BlYxIRCglkaXNrX3R5cGUYASABKAkSFAoMZGlza19zaXplX2diGAIgASgD",
55-
"IkwKCE5mc01vdW50EhMKBnNlcnZlchgBIAEoCUID4EECEhEKBHBhdGgYAiAB",
56-
"KAlCA+BBAhIYCgttb3VudF9wb2ludBgDIAEoCUID4EECIkEKFUF1dG9zY2Fs",
57-
"aW5nTWV0cmljU3BlYxIYCgttZXRyaWNfbmFtZRgBIAEoCUID4EECEg4KBnRh",
58-
"cmdldBgCIAEoBSIuChBTaGllbGRlZFZtQ29uZmlnEhoKEmVuYWJsZV9zZWN1",
59-
"cmVfYm9vdBgBIAEoCELsAQojY29tLmdvb2dsZS5jbG91ZC5haXBsYXRmb3Jt",
60-
"LnYxYmV0YTFCFU1hY2hpbmVSZXNvdXJjZXNQcm90b1ABWkNjbG91ZC5nb29n",
61-
"bGUuY29tL2dvL2FpcGxhdGZvcm0vYXBpdjFiZXRhMS9haXBsYXRmb3JtcGI7",
62-
"YWlwbGF0Zm9ybXBiqgIfR29vZ2xlLkNsb3VkLkFJUGxhdGZvcm0uVjFCZXRh",
63-
"McoCH0dvb2dsZVxDbG91ZFxBSVBsYXRmb3JtXFYxYmV0YTHqAiJHb29nbGU6",
64-
"OkNsb3VkOjpBSVBsYXRmb3JtOjpWMWJldGExYgZwcm90bzM="));
50+
"dGExLk1hY2hpbmVTcGVjQgbgQQLgQQUSIwoWc3RhcnRpbmdfcmVwbGljYV9j",
51+
"b3VudBgCIAEoBUID4EEFEh4KEW1heF9yZXBsaWNhX2NvdW50GAMgASgFQgPg",
52+
"QQUiLwoRUmVzb3VyY2VzQ29uc3VtZWQSGgoNcmVwbGljYV9ob3VycxgBIAEo",
53+
"AUID4EEDIj0KCERpc2tTcGVjEhYKDmJvb3RfZGlza190eXBlGAEgASgJEhkK",
54+
"EWJvb3RfZGlza19zaXplX2diGAIgASgFIj0KElBlcnNpc3RlbnREaXNrU3Bl",
55+
"YxIRCglkaXNrX3R5cGUYASABKAkSFAoMZGlza19zaXplX2diGAIgASgDIkwK",
56+
"CE5mc01vdW50EhMKBnNlcnZlchgBIAEoCUID4EECEhEKBHBhdGgYAiABKAlC",
57+
"A+BBAhIYCgttb3VudF9wb2ludBgDIAEoCUID4EECIkEKFUF1dG9zY2FsaW5n",
58+
"TWV0cmljU3BlYxIYCgttZXRyaWNfbmFtZRgBIAEoCUID4EECEg4KBnRhcmdl",
59+
"dBgCIAEoBSIuChBTaGllbGRlZFZtQ29uZmlnEhoKEmVuYWJsZV9zZWN1cmVf",
60+
"Ym9vdBgBIAEoCELsAQojY29tLmdvb2dsZS5jbG91ZC5haXBsYXRmb3JtLnYx",
61+
"YmV0YTFCFU1hY2hpbmVSZXNvdXJjZXNQcm90b1ABWkNjbG91ZC5nb29nbGUu",
62+
"Y29tL2dvL2FpcGxhdGZvcm0vYXBpdjFiZXRhMS9haXBsYXRmb3JtcGI7YWlw",
63+
"bGF0Zm9ybXBiqgIfR29vZ2xlLkNsb3VkLkFJUGxhdGZvcm0uVjFCZXRhMcoC",
64+
"H0dvb2dsZVxDbG91ZFxBSVBsYXRmb3JtXFYxYmV0YTHqAiJHb29nbGU6OkNs",
65+
"b3VkOjpBSVBsYXRmb3JtOjpWMWJldGExYgZwcm90bzM="));
6566
descriptor = pbr::FileDescriptor.FromGeneratedCode(descriptorData,
6667
new pbr::FileDescriptor[] { global::Google.Api.FieldBehaviorReflection.Descriptor, global::Google.Cloud.AIPlatform.V1Beta1.AcceleratorTypeReflection.Descriptor, global::Google.Cloud.AIPlatform.V1Beta1.ReservationAffinityReflection.Descriptor, },
6768
new pbr::GeneratedClrTypeInfo(null, null, new pbr::GeneratedClrTypeInfo[] {
68-
new pbr::GeneratedClrTypeInfo(typeof(global::Google.Cloud.AIPlatform.V1Beta1.MachineSpec), global::Google.Cloud.AIPlatform.V1Beta1.MachineSpec.Parser, new[]{ "MachineType", "AcceleratorType", "AcceleratorCount", "TpuTopology", "ReservationAffinity" }, null, null, null, null),
69+
new pbr::GeneratedClrTypeInfo(typeof(global::Google.Cloud.AIPlatform.V1Beta1.MachineSpec), global::Google.Cloud.AIPlatform.V1Beta1.MachineSpec.Parser, new[]{ "MachineType", "AcceleratorType", "AcceleratorCount", "TpuTopology", "MultihostGpuNodeCount", "ReservationAffinity" }, null, null, null, null),
6970
new pbr::GeneratedClrTypeInfo(typeof(global::Google.Cloud.AIPlatform.V1Beta1.DedicatedResources), global::Google.Cloud.AIPlatform.V1Beta1.DedicatedResources.Parser, new[]{ "MachineSpec", "MinReplicaCount", "MaxReplicaCount", "RequiredReplicaCount", "AutoscalingMetricSpecs", "Spot" }, null, null, null, null),
7071
new pbr::GeneratedClrTypeInfo(typeof(global::Google.Cloud.AIPlatform.V1Beta1.AutomaticResources), global::Google.Cloud.AIPlatform.V1Beta1.AutomaticResources.Parser, new[]{ "MinReplicaCount", "MaxReplicaCount" }, null, null, null, null),
7172
new pbr::GeneratedClrTypeInfo(typeof(global::Google.Cloud.AIPlatform.V1Beta1.BatchDedicatedResources), global::Google.Cloud.AIPlatform.V1Beta1.BatchDedicatedResources.Parser, new[]{ "MachineSpec", "StartingReplicaCount", "MaxReplicaCount" }, null, null, null, null),
@@ -123,6 +124,7 @@ public MachineSpec(MachineSpec other) : this() {
123124
acceleratorType_ = other.acceleratorType_;
124125
acceleratorCount_ = other.acceleratorCount_;
125126
tpuTopology_ = other.tpuTopology_;
127+
multihostGpuNodeCount_ = other.multihostGpuNodeCount_;
126128
reservationAffinity_ = other.reservationAffinity_ != null ? other.reservationAffinity_.Clone() : null;
127129
_unknownFields = pb::UnknownFieldSet.Clone(other._unknownFields);
128130
}
@@ -208,6 +210,22 @@ public string TpuTopology {
208210
}
209211
}
210212

213+
/// <summary>Field number for the "multihost_gpu_node_count" field.</summary>
214+
public const int MultihostGpuNodeCountFieldNumber = 6;
215+
private int multihostGpuNodeCount_;
216+
/// <summary>
217+
/// Optional. Immutable. The number of nodes per replica for multihost GPU
218+
/// deployments.
219+
/// </summary>
220+
[global::System.Diagnostics.DebuggerNonUserCodeAttribute]
221+
[global::System.CodeDom.Compiler.GeneratedCode("protoc", null)]
222+
public int MultihostGpuNodeCount {
223+
get { return multihostGpuNodeCount_; }
224+
set {
225+
multihostGpuNodeCount_ = value;
226+
}
227+
}
228+
211229
/// <summary>Field number for the "reservation_affinity" field.</summary>
212230
public const int ReservationAffinityFieldNumber = 5;
213231
private global::Google.Cloud.AIPlatform.V1Beta1.ReservationAffinity reservationAffinity_;
@@ -243,6 +261,7 @@ public bool Equals(MachineSpec other) {
243261
if (AcceleratorType != other.AcceleratorType) return false;
244262
if (AcceleratorCount != other.AcceleratorCount) return false;
245263
if (TpuTopology != other.TpuTopology) return false;
264+
if (MultihostGpuNodeCount != other.MultihostGpuNodeCount) return false;
246265
if (!object.Equals(ReservationAffinity, other.ReservationAffinity)) return false;
247266
return Equals(_unknownFields, other._unknownFields);
248267
}
@@ -255,6 +274,7 @@ public override int GetHashCode() {
255274
if (AcceleratorType != global::Google.Cloud.AIPlatform.V1Beta1.AcceleratorType.Unspecified) hash ^= AcceleratorType.GetHashCode();
256275
if (AcceleratorCount != 0) hash ^= AcceleratorCount.GetHashCode();
257276
if (TpuTopology.Length != 0) hash ^= TpuTopology.GetHashCode();
277+
if (MultihostGpuNodeCount != 0) hash ^= MultihostGpuNodeCount.GetHashCode();
258278
if (reservationAffinity_ != null) hash ^= ReservationAffinity.GetHashCode();
259279
if (_unknownFields != null) {
260280
hash ^= _unknownFields.GetHashCode();
@@ -294,6 +314,10 @@ public void WriteTo(pb::CodedOutputStream output) {
294314
output.WriteRawTag(42);
295315
output.WriteMessage(ReservationAffinity);
296316
}
317+
if (MultihostGpuNodeCount != 0) {
318+
output.WriteRawTag(48);
319+
output.WriteInt32(MultihostGpuNodeCount);
320+
}
297321
if (_unknownFields != null) {
298322
_unknownFields.WriteTo(output);
299323
}
@@ -324,6 +348,10 @@ public void WriteTo(pb::CodedOutputStream output) {
324348
output.WriteRawTag(42);
325349
output.WriteMessage(ReservationAffinity);
326350
}
351+
if (MultihostGpuNodeCount != 0) {
352+
output.WriteRawTag(48);
353+
output.WriteInt32(MultihostGpuNodeCount);
354+
}
327355
if (_unknownFields != null) {
328356
_unknownFields.WriteTo(ref output);
329357
}
@@ -346,6 +374,9 @@ public int CalculateSize() {
346374
if (TpuTopology.Length != 0) {
347375
size += 1 + pb::CodedOutputStream.ComputeStringSize(TpuTopology);
348376
}
377+
if (MultihostGpuNodeCount != 0) {
378+
size += 1 + pb::CodedOutputStream.ComputeInt32Size(MultihostGpuNodeCount);
379+
}
349380
if (reservationAffinity_ != null) {
350381
size += 1 + pb::CodedOutputStream.ComputeMessageSize(ReservationAffinity);
351382
}
@@ -373,6 +404,9 @@ public void MergeFrom(MachineSpec other) {
373404
if (other.TpuTopology.Length != 0) {
374405
TpuTopology = other.TpuTopology;
375406
}
407+
if (other.MultihostGpuNodeCount != 0) {
408+
MultihostGpuNodeCount = other.MultihostGpuNodeCount;
409+
}
376410
if (other.reservationAffinity_ != null) {
377411
if (reservationAffinity_ == null) {
378412
ReservationAffinity = new global::Google.Cloud.AIPlatform.V1Beta1.ReservationAffinity();
@@ -417,6 +451,10 @@ public void MergeFrom(pb::CodedInputStream input) {
417451
input.ReadMessage(ReservationAffinity);
418452
break;
419453
}
454+
case 48: {
455+
MultihostGpuNodeCount = input.ReadInt32();
456+
break;
457+
}
420458
}
421459
}
422460
#endif
@@ -455,6 +493,10 @@ public void MergeFrom(pb::CodedInputStream input) {
455493
input.ReadMessage(ReservationAffinity);
456494
break;
457495
}
496+
case 48: {
497+
MultihostGpuNodeCount = input.ReadInt32();
498+
break;
499+
}
458500
}
459501
}
460502
}

0 commit comments

Comments
 (0)