@flyteorg/flyteidl 1.5.16 → 1.5.18

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1886,6 +1886,9 @@ export namespace flyteidl {
1886
1886
 
1887
1887
  /** TaskNodeOverrides resources */
1888
1888
  resources?: (flyteidl.core.IResources|null);
1889
+
1890
+ /** TaskNodeOverrides resourceMetadata */
1891
+ resourceMetadata?: (flyteidl.core.IResourceMetadata|null);
1889
1892
  }
1890
1893
 
1891
1894
  /** Represents a TaskNodeOverrides. */
@@ -1900,6 +1903,9 @@ export namespace flyteidl {
1900
1903
  /** TaskNodeOverrides resources. */
1901
1904
  public resources?: (flyteidl.core.IResources|null);
1902
1905
 
1906
+ /** TaskNodeOverrides resourceMetadata. */
1907
+ public resourceMetadata?: (flyteidl.core.IResourceMetadata|null);
1908
+
1903
1909
  /**
1904
1910
  * Creates a new TaskNodeOverrides instance using the specified properties.
1905
1911
  * @param [properties] Properties to set
@@ -5157,6 +5163,128 @@ export namespace flyteidl {
5157
5163
  }
5158
5164
  }
5159
5165
 
5166
+ /** Properties of a GPUAccelerator. */
5167
+ interface IGPUAccelerator {
5168
+
5169
+ /** GPUAccelerator device */
5170
+ device?: (string|null);
5171
+
5172
+ /** GPUAccelerator unpartitioned */
5173
+ unpartitioned?: (boolean|null);
5174
+
5175
+ /** GPUAccelerator partitionSize */
5176
+ partitionSize?: (string|null);
5177
+ }
5178
+
5179
+ /** Represents a GPUAccelerator. */
5180
+ class GPUAccelerator implements IGPUAccelerator {
5181
+
5182
+ /**
5183
+ * Constructs a new GPUAccelerator.
5184
+ * @param [properties] Properties to set
5185
+ */
5186
+ constructor(properties?: flyteidl.core.IGPUAccelerator);
5187
+
5188
+ /** GPUAccelerator device. */
5189
+ public device: string;
5190
+
5191
+ /** GPUAccelerator unpartitioned. */
5192
+ public unpartitioned: boolean;
5193
+
5194
+ /** GPUAccelerator partitionSize. */
5195
+ public partitionSize: string;
5196
+
5197
+ /** GPUAccelerator partitionSizeValue. */
5198
+ public partitionSizeValue?: ("unpartitioned"|"partitionSize");
5199
+
5200
+ /**
5201
+ * Creates a new GPUAccelerator instance using the specified properties.
5202
+ * @param [properties] Properties to set
5203
+ * @returns GPUAccelerator instance
5204
+ */
5205
+ public static create(properties?: flyteidl.core.IGPUAccelerator): flyteidl.core.GPUAccelerator;
5206
+
5207
+ /**
5208
+ * Encodes the specified GPUAccelerator message. Does not implicitly {@link flyteidl.core.GPUAccelerator.verify|verify} messages.
5209
+ * @param message GPUAccelerator message or plain object to encode
5210
+ * @param [writer] Writer to encode to
5211
+ * @returns Writer
5212
+ */
5213
+ public static encode(message: flyteidl.core.IGPUAccelerator, writer?: $protobuf.Writer): $protobuf.Writer;
5214
+
5215
+ /**
5216
+ * Decodes a GPUAccelerator message from the specified reader or buffer.
5217
+ * @param reader Reader or buffer to decode from
5218
+ * @param [length] Message length if known beforehand
5219
+ * @returns GPUAccelerator
5220
+ * @throws {Error} If the payload is not a reader or valid buffer
5221
+ * @throws {$protobuf.util.ProtocolError} If required fields are missing
5222
+ */
5223
+ public static decode(reader: ($protobuf.Reader|Uint8Array), length?: number): flyteidl.core.GPUAccelerator;
5224
+
5225
+ /**
5226
+ * Verifies a GPUAccelerator message.
5227
+ * @param message Plain object to verify
5228
+ * @returns `null` if valid, otherwise the reason why it is not
5229
+ */
5230
+ public static verify(message: { [k: string]: any }): (string|null);
5231
+ }
5232
+
5233
+ /** Properties of a ResourceMetadata. */
5234
+ interface IResourceMetadata {
5235
+
5236
+ /** ResourceMetadata gpuAccelerator */
5237
+ gpuAccelerator?: (flyteidl.core.IGPUAccelerator|null);
5238
+ }
5239
+
5240
+ /** Represents a ResourceMetadata. */
5241
+ class ResourceMetadata implements IResourceMetadata {
5242
+
5243
+ /**
5244
+ * Constructs a new ResourceMetadata.
5245
+ * @param [properties] Properties to set
5246
+ */
5247
+ constructor(properties?: flyteidl.core.IResourceMetadata);
5248
+
5249
+ /** ResourceMetadata gpuAccelerator. */
5250
+ public gpuAccelerator?: (flyteidl.core.IGPUAccelerator|null);
5251
+
5252
+ /** ResourceMetadata acceleratorValue. */
5253
+ public acceleratorValue?: "gpuAccelerator";
5254
+
5255
+ /**
5256
+ * Creates a new ResourceMetadata instance using the specified properties.
5257
+ * @param [properties] Properties to set
5258
+ * @returns ResourceMetadata instance
5259
+ */
5260
+ public static create(properties?: flyteidl.core.IResourceMetadata): flyteidl.core.ResourceMetadata;
5261
+
5262
+ /**
5263
+ * Encodes the specified ResourceMetadata message. Does not implicitly {@link flyteidl.core.ResourceMetadata.verify|verify} messages.
5264
+ * @param message ResourceMetadata message or plain object to encode
5265
+ * @param [writer] Writer to encode to
5266
+ * @returns Writer
5267
+ */
5268
+ public static encode(message: flyteidl.core.IResourceMetadata, writer?: $protobuf.Writer): $protobuf.Writer;
5269
+
5270
+ /**
5271
+ * Decodes a ResourceMetadata message from the specified reader or buffer.
5272
+ * @param reader Reader or buffer to decode from
5273
+ * @param [length] Message length if known beforehand
5274
+ * @returns ResourceMetadata
5275
+ * @throws {Error} If the payload is not a reader or valid buffer
5276
+ * @throws {$protobuf.util.ProtocolError} If required fields are missing
5277
+ */
5278
+ public static decode(reader: ($protobuf.Reader|Uint8Array), length?: number): flyteidl.core.ResourceMetadata;
5279
+
5280
+ /**
5281
+ * Verifies a ResourceMetadata message.
5282
+ * @param message Plain object to verify
5283
+ * @returns `null` if valid, otherwise the reason why it is not
5284
+ */
5285
+ public static verify(message: { [k: string]: any }): (string|null);
5286
+ }
5287
+
5160
5288
  /** Properties of a TaskMetadata. */
5161
5289
  interface ITaskMetadata {
5162
5290
 
@@ -5192,6 +5320,9 @@ export namespace flyteidl {
5192
5320
 
5193
5321
  /** TaskMetadata podTemplateName */
5194
5322
  podTemplateName?: (string|null);
5323
+
5324
+ /** TaskMetadata resourceMetadata */
5325
+ resourceMetadata?: (flyteidl.core.IResourceMetadata|null);
5195
5326
  }
5196
5327
 
5197
5328
  /** Represents a TaskMetadata. */
@@ -5236,6 +5367,9 @@ export namespace flyteidl {
5236
5367
  /** TaskMetadata podTemplateName. */
5237
5368
  public podTemplateName: string;
5238
5369
 
5370
+ /** TaskMetadata resourceMetadata. */
5371
+ public resourceMetadata?: (flyteidl.core.IResourceMetadata|null);
5372
+
5239
5373
  /** TaskMetadata interruptibleValue. */
5240
5374
  public interruptibleValue?: "interruptible";
5241
5375
 
@@ -4570,6 +4570,7 @@
4570
4570
  * @memberof flyteidl.core
4571
4571
  * @interface ITaskNodeOverrides
4572
4572
  * @property {flyteidl.core.IResources|null} [resources] TaskNodeOverrides resources
4573
+ * @property {flyteidl.core.IResourceMetadata|null} [resourceMetadata] TaskNodeOverrides resourceMetadata
4573
4574
  */
4574
4575
 
4575
4576
  /**
@@ -4595,6 +4596,14 @@
4595
4596
  */
4596
4597
  TaskNodeOverrides.prototype.resources = null;
4597
4598
 
4599
+ /**
4600
+ * TaskNodeOverrides resourceMetadata.
4601
+ * @member {flyteidl.core.IResourceMetadata|null|undefined} resourceMetadata
4602
+ * @memberof flyteidl.core.TaskNodeOverrides
4603
+ * @instance
4604
+ */
4605
+ TaskNodeOverrides.prototype.resourceMetadata = null;
4606
+
4598
4607
  /**
4599
4608
  * Creates a new TaskNodeOverrides instance using the specified properties.
4600
4609
  * @function create
@@ -4621,6 +4630,8 @@
4621
4630
  writer = $Writer.create();
4622
4631
  if (message.resources != null && message.hasOwnProperty("resources"))
4623
4632
  $root.flyteidl.core.Resources.encode(message.resources, writer.uint32(/* id 1, wireType 2 =*/10).fork()).ldelim();
4633
+ if (message.resourceMetadata != null && message.hasOwnProperty("resourceMetadata"))
4634
+ $root.flyteidl.core.ResourceMetadata.encode(message.resourceMetadata, writer.uint32(/* id 2, wireType 2 =*/18).fork()).ldelim();
4624
4635
  return writer;
4625
4636
  };
4626
4637
 
@@ -4645,6 +4656,9 @@
4645
4656
  case 1:
4646
4657
  message.resources = $root.flyteidl.core.Resources.decode(reader, reader.uint32());
4647
4658
  break;
4659
+ case 2:
4660
+ message.resourceMetadata = $root.flyteidl.core.ResourceMetadata.decode(reader, reader.uint32());
4661
+ break;
4648
4662
  default:
4649
4663
  reader.skipType(tag & 7);
4650
4664
  break;
@@ -4669,6 +4683,11 @@
4669
4683
  if (error)
4670
4684
  return "resources." + error;
4671
4685
  }
4686
+ if (message.resourceMetadata != null && message.hasOwnProperty("resourceMetadata")) {
4687
+ var error = $root.flyteidl.core.ResourceMetadata.verify(message.resourceMetadata);
4688
+ if (error)
4689
+ return "resourceMetadata." + error;
4690
+ }
4672
4691
  return null;
4673
4692
  };
4674
4693
 
@@ -12264,6 +12283,301 @@
12264
12283
  return RuntimeMetadata;
12265
12284
  })();
12266
12285
 
12286
+ core.GPUAccelerator = (function() {
12287
+
12288
+ /**
12289
+ * Properties of a GPUAccelerator.
12290
+ * @memberof flyteidl.core
12291
+ * @interface IGPUAccelerator
12292
+ * @property {string|null} [device] GPUAccelerator device
12293
+ * @property {boolean|null} [unpartitioned] GPUAccelerator unpartitioned
12294
+ * @property {string|null} [partitionSize] GPUAccelerator partitionSize
12295
+ */
12296
+
12297
+ /**
12298
+ * Constructs a new GPUAccelerator.
12299
+ * @memberof flyteidl.core
12300
+ * @classdesc Represents a GPUAccelerator.
12301
+ * @implements IGPUAccelerator
12302
+ * @constructor
12303
+ * @param {flyteidl.core.IGPUAccelerator=} [properties] Properties to set
12304
+ */
12305
+ function GPUAccelerator(properties) {
12306
+ if (properties)
12307
+ for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
12308
+ if (properties[keys[i]] != null)
12309
+ this[keys[i]] = properties[keys[i]];
12310
+ }
12311
+
12312
+ /**
12313
+ * GPUAccelerator device.
12314
+ * @member {string} device
12315
+ * @memberof flyteidl.core.GPUAccelerator
12316
+ * @instance
12317
+ */
12318
+ GPUAccelerator.prototype.device = "";
12319
+
12320
+ /**
12321
+ * GPUAccelerator unpartitioned.
12322
+ * @member {boolean} unpartitioned
12323
+ * @memberof flyteidl.core.GPUAccelerator
12324
+ * @instance
12325
+ */
12326
+ GPUAccelerator.prototype.unpartitioned = false;
12327
+
12328
+ /**
12329
+ * GPUAccelerator partitionSize.
12330
+ * @member {string} partitionSize
12331
+ * @memberof flyteidl.core.GPUAccelerator
12332
+ * @instance
12333
+ */
12334
+ GPUAccelerator.prototype.partitionSize = "";
12335
+
12336
+ // OneOf field names bound to virtual getters and setters
12337
+ var $oneOfFields;
12338
+
12339
+ /**
12340
+ * GPUAccelerator partitionSizeValue.
12341
+ * @member {"unpartitioned"|"partitionSize"|undefined} partitionSizeValue
12342
+ * @memberof flyteidl.core.GPUAccelerator
12343
+ * @instance
12344
+ */
12345
+ Object.defineProperty(GPUAccelerator.prototype, "partitionSizeValue", {
12346
+ get: $util.oneOfGetter($oneOfFields = ["unpartitioned", "partitionSize"]),
12347
+ set: $util.oneOfSetter($oneOfFields)
12348
+ });
12349
+
12350
+ /**
12351
+ * Creates a new GPUAccelerator instance using the specified properties.
12352
+ * @function create
12353
+ * @memberof flyteidl.core.GPUAccelerator
12354
+ * @static
12355
+ * @param {flyteidl.core.IGPUAccelerator=} [properties] Properties to set
12356
+ * @returns {flyteidl.core.GPUAccelerator} GPUAccelerator instance
12357
+ */
12358
+ GPUAccelerator.create = function create(properties) {
12359
+ return new GPUAccelerator(properties);
12360
+ };
12361
+
12362
+ /**
12363
+ * Encodes the specified GPUAccelerator message. Does not implicitly {@link flyteidl.core.GPUAccelerator.verify|verify} messages.
12364
+ * @function encode
12365
+ * @memberof flyteidl.core.GPUAccelerator
12366
+ * @static
12367
+ * @param {flyteidl.core.IGPUAccelerator} message GPUAccelerator message or plain object to encode
12368
+ * @param {$protobuf.Writer} [writer] Writer to encode to
12369
+ * @returns {$protobuf.Writer} Writer
12370
+ */
12371
+ GPUAccelerator.encode = function encode(message, writer) {
12372
+ if (!writer)
12373
+ writer = $Writer.create();
12374
+ if (message.device != null && message.hasOwnProperty("device"))
12375
+ writer.uint32(/* id 1, wireType 2 =*/10).string(message.device);
12376
+ if (message.unpartitioned != null && message.hasOwnProperty("unpartitioned"))
12377
+ writer.uint32(/* id 2, wireType 0 =*/16).bool(message.unpartitioned);
12378
+ if (message.partitionSize != null && message.hasOwnProperty("partitionSize"))
12379
+ writer.uint32(/* id 3, wireType 2 =*/26).string(message.partitionSize);
12380
+ return writer;
12381
+ };
12382
+
12383
+ /**
12384
+ * Decodes a GPUAccelerator message from the specified reader or buffer.
12385
+ * @function decode
12386
+ * @memberof flyteidl.core.GPUAccelerator
12387
+ * @static
12388
+ * @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
12389
+ * @param {number} [length] Message length if known beforehand
12390
+ * @returns {flyteidl.core.GPUAccelerator} GPUAccelerator
12391
+ * @throws {Error} If the payload is not a reader or valid buffer
12392
+ * @throws {$protobuf.util.ProtocolError} If required fields are missing
12393
+ */
12394
+ GPUAccelerator.decode = function decode(reader, length) {
12395
+ if (!(reader instanceof $Reader))
12396
+ reader = $Reader.create(reader);
12397
+ var end = length === undefined ? reader.len : reader.pos + length, message = new $root.flyteidl.core.GPUAccelerator();
12398
+ while (reader.pos < end) {
12399
+ var tag = reader.uint32();
12400
+ switch (tag >>> 3) {
12401
+ case 1:
12402
+ message.device = reader.string();
12403
+ break;
12404
+ case 2:
12405
+ message.unpartitioned = reader.bool();
12406
+ break;
12407
+ case 3:
12408
+ message.partitionSize = reader.string();
12409
+ break;
12410
+ default:
12411
+ reader.skipType(tag & 7);
12412
+ break;
12413
+ }
12414
+ }
12415
+ return message;
12416
+ };
12417
+
12418
+ /**
12419
+ * Verifies a GPUAccelerator message.
12420
+ * @function verify
12421
+ * @memberof flyteidl.core.GPUAccelerator
12422
+ * @static
12423
+ * @param {Object.<string,*>} message Plain object to verify
12424
+ * @returns {string|null} `null` if valid, otherwise the reason why it is not
12425
+ */
12426
+ GPUAccelerator.verify = function verify(message) {
12427
+ if (typeof message !== "object" || message === null)
12428
+ return "object expected";
12429
+ var properties = {};
12430
+ if (message.device != null && message.hasOwnProperty("device"))
12431
+ if (!$util.isString(message.device))
12432
+ return "device: string expected";
12433
+ if (message.unpartitioned != null && message.hasOwnProperty("unpartitioned")) {
12434
+ properties.partitionSizeValue = 1;
12435
+ if (typeof message.unpartitioned !== "boolean")
12436
+ return "unpartitioned: boolean expected";
12437
+ }
12438
+ if (message.partitionSize != null && message.hasOwnProperty("partitionSize")) {
12439
+ if (properties.partitionSizeValue === 1)
12440
+ return "partitionSizeValue: multiple values";
12441
+ properties.partitionSizeValue = 1;
12442
+ if (!$util.isString(message.partitionSize))
12443
+ return "partitionSize: string expected";
12444
+ }
12445
+ return null;
12446
+ };
12447
+
12448
+ return GPUAccelerator;
12449
+ })();
12450
+
12451
+ core.ResourceMetadata = (function() {
12452
+
12453
+ /**
12454
+ * Properties of a ResourceMetadata.
12455
+ * @memberof flyteidl.core
12456
+ * @interface IResourceMetadata
12457
+ * @property {flyteidl.core.IGPUAccelerator|null} [gpuAccelerator] ResourceMetadata gpuAccelerator
12458
+ */
12459
+
12460
+ /**
12461
+ * Constructs a new ResourceMetadata.
12462
+ * @memberof flyteidl.core
12463
+ * @classdesc Represents a ResourceMetadata.
12464
+ * @implements IResourceMetadata
12465
+ * @constructor
12466
+ * @param {flyteidl.core.IResourceMetadata=} [properties] Properties to set
12467
+ */
12468
+ function ResourceMetadata(properties) {
12469
+ if (properties)
12470
+ for (var keys = Object.keys(properties), i = 0; i < keys.length; ++i)
12471
+ if (properties[keys[i]] != null)
12472
+ this[keys[i]] = properties[keys[i]];
12473
+ }
12474
+
12475
+ /**
12476
+ * ResourceMetadata gpuAccelerator.
12477
+ * @member {flyteidl.core.IGPUAccelerator|null|undefined} gpuAccelerator
12478
+ * @memberof flyteidl.core.ResourceMetadata
12479
+ * @instance
12480
+ */
12481
+ ResourceMetadata.prototype.gpuAccelerator = null;
12482
+
12483
+ // OneOf field names bound to virtual getters and setters
12484
+ var $oneOfFields;
12485
+
12486
+ /**
12487
+ * ResourceMetadata acceleratorValue.
12488
+ * @member {"gpuAccelerator"|undefined} acceleratorValue
12489
+ * @memberof flyteidl.core.ResourceMetadata
12490
+ * @instance
12491
+ */
12492
+ Object.defineProperty(ResourceMetadata.prototype, "acceleratorValue", {
12493
+ get: $util.oneOfGetter($oneOfFields = ["gpuAccelerator"]),
12494
+ set: $util.oneOfSetter($oneOfFields)
12495
+ });
12496
+
12497
+ /**
12498
+ * Creates a new ResourceMetadata instance using the specified properties.
12499
+ * @function create
12500
+ * @memberof flyteidl.core.ResourceMetadata
12501
+ * @static
12502
+ * @param {flyteidl.core.IResourceMetadata=} [properties] Properties to set
12503
+ * @returns {flyteidl.core.ResourceMetadata} ResourceMetadata instance
12504
+ */
12505
+ ResourceMetadata.create = function create(properties) {
12506
+ return new ResourceMetadata(properties);
12507
+ };
12508
+
12509
+ /**
12510
+ * Encodes the specified ResourceMetadata message. Does not implicitly {@link flyteidl.core.ResourceMetadata.verify|verify} messages.
12511
+ * @function encode
12512
+ * @memberof flyteidl.core.ResourceMetadata
12513
+ * @static
12514
+ * @param {flyteidl.core.IResourceMetadata} message ResourceMetadata message or plain object to encode
12515
+ * @param {$protobuf.Writer} [writer] Writer to encode to
12516
+ * @returns {$protobuf.Writer} Writer
12517
+ */
12518
+ ResourceMetadata.encode = function encode(message, writer) {
12519
+ if (!writer)
12520
+ writer = $Writer.create();
12521
+ if (message.gpuAccelerator != null && message.hasOwnProperty("gpuAccelerator"))
12522
+ $root.flyteidl.core.GPUAccelerator.encode(message.gpuAccelerator, writer.uint32(/* id 1, wireType 2 =*/10).fork()).ldelim();
12523
+ return writer;
12524
+ };
12525
+
12526
+ /**
12527
+ * Decodes a ResourceMetadata message from the specified reader or buffer.
12528
+ * @function decode
12529
+ * @memberof flyteidl.core.ResourceMetadata
12530
+ * @static
12531
+ * @param {$protobuf.Reader|Uint8Array} reader Reader or buffer to decode from
12532
+ * @param {number} [length] Message length if known beforehand
12533
+ * @returns {flyteidl.core.ResourceMetadata} ResourceMetadata
12534
+ * @throws {Error} If the payload is not a reader or valid buffer
12535
+ * @throws {$protobuf.util.ProtocolError} If required fields are missing
12536
+ */
12537
+ ResourceMetadata.decode = function decode(reader, length) {
12538
+ if (!(reader instanceof $Reader))
12539
+ reader = $Reader.create(reader);
12540
+ var end = length === undefined ? reader.len : reader.pos + length, message = new $root.flyteidl.core.ResourceMetadata();
12541
+ while (reader.pos < end) {
12542
+ var tag = reader.uint32();
12543
+ switch (tag >>> 3) {
12544
+ case 1:
12545
+ message.gpuAccelerator = $root.flyteidl.core.GPUAccelerator.decode(reader, reader.uint32());
12546
+ break;
12547
+ default:
12548
+ reader.skipType(tag & 7);
12549
+ break;
12550
+ }
12551
+ }
12552
+ return message;
12553
+ };
12554
+
12555
+ /**
12556
+ * Verifies a ResourceMetadata message.
12557
+ * @function verify
12558
+ * @memberof flyteidl.core.ResourceMetadata
12559
+ * @static
12560
+ * @param {Object.<string,*>} message Plain object to verify
12561
+ * @returns {string|null} `null` if valid, otherwise the reason why it is not
12562
+ */
12563
+ ResourceMetadata.verify = function verify(message) {
12564
+ if (typeof message !== "object" || message === null)
12565
+ return "object expected";
12566
+ var properties = {};
12567
+ if (message.gpuAccelerator != null && message.hasOwnProperty("gpuAccelerator")) {
12568
+ properties.acceleratorValue = 1;
12569
+ {
12570
+ var error = $root.flyteidl.core.GPUAccelerator.verify(message.gpuAccelerator);
12571
+ if (error)
12572
+ return "gpuAccelerator." + error;
12573
+ }
12574
+ }
12575
+ return null;
12576
+ };
12577
+
12578
+ return ResourceMetadata;
12579
+ })();
12580
+
12267
12581
  core.TaskMetadata = (function() {
12268
12582
 
12269
12583
  /**
@@ -12281,6 +12595,7 @@
12281
12595
  * @property {boolean|null} [generatesDeck] TaskMetadata generatesDeck
12282
12596
  * @property {Object.<string,string>|null} [tags] TaskMetadata tags
12283
12597
  * @property {string|null} [podTemplateName] TaskMetadata podTemplateName
12598
+ * @property {flyteidl.core.IResourceMetadata|null} [resourceMetadata] TaskMetadata resourceMetadata
12284
12599
  */
12285
12600
 
12286
12601
  /**
@@ -12387,6 +12702,14 @@
12387
12702
  */
12388
12703
  TaskMetadata.prototype.podTemplateName = "";
12389
12704
 
12705
+ /**
12706
+ * TaskMetadata resourceMetadata.
12707
+ * @member {flyteidl.core.IResourceMetadata|null|undefined} resourceMetadata
12708
+ * @memberof flyteidl.core.TaskMetadata
12709
+ * @instance
12710
+ */
12711
+ TaskMetadata.prototype.resourceMetadata = null;
12712
+
12390
12713
  // OneOf field names bound to virtual getters and setters
12391
12714
  var $oneOfFields;
12392
12715
 
@@ -12448,6 +12771,8 @@
12448
12771
  writer.uint32(/* id 11, wireType 2 =*/90).fork().uint32(/* id 1, wireType 2 =*/10).string(keys[i]).uint32(/* id 2, wireType 2 =*/18).string(message.tags[keys[i]]).ldelim();
12449
12772
  if (message.podTemplateName != null && message.hasOwnProperty("podTemplateName"))
12450
12773
  writer.uint32(/* id 12, wireType 2 =*/98).string(message.podTemplateName);
12774
+ if (message.resourceMetadata != null && message.hasOwnProperty("resourceMetadata"))
12775
+ $root.flyteidl.core.ResourceMetadata.encode(message.resourceMetadata, writer.uint32(/* id 13, wireType 2 =*/106).fork()).ldelim();
12451
12776
  return writer;
12452
12777
  };
12453
12778
 
@@ -12507,6 +12832,9 @@
12507
12832
  case 12:
12508
12833
  message.podTemplateName = reader.string();
12509
12834
  break;
12835
+ case 13:
12836
+ message.resourceMetadata = $root.flyteidl.core.ResourceMetadata.decode(reader, reader.uint32());
12837
+ break;
12510
12838
  default:
12511
12839
  reader.skipType(tag & 7);
12512
12840
  break;
@@ -12573,6 +12901,11 @@
12573
12901
  if (message.podTemplateName != null && message.hasOwnProperty("podTemplateName"))
12574
12902
  if (!$util.isString(message.podTemplateName))
12575
12903
  return "podTemplateName: string expected";
12904
+ if (message.resourceMetadata != null && message.hasOwnProperty("resourceMetadata")) {
12905
+ var error = $root.flyteidl.core.ResourceMetadata.verify(message.resourceMetadata);
12906
+ if (error)
12907
+ return "resourceMetadata." + error;
12908
+ }
12576
12909
  return null;
12577
12910
  };
12578
12911
 
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@flyteorg/flyteidl",
3
- "version": "1.5.16",
3
+ "version": "1.5.18",
4
4
  "description": "Compiled protocol buffers and gRPC service clients/servers for Flyte IDLs",
5
5
  "repository": {
6
6
  "type": "git",
@@ -61,6 +61,22 @@ message RuntimeMetadata {
61
61
  string flavor = 3;
62
62
  }
63
63
 
64
+ // Metadata associated with the GPU accelerator to allocate to a task
65
+ message GPUAccelerator {
66
+ string device = 1;
67
+ oneof partition_size_value {
68
+ bool unpartitioned = 2;
69
+ string partition_size = 3;
70
+ }
71
+ }
72
+
73
+ // Additional metadata associated with resources to allocate to a task
74
+ message ResourceMetadata {
75
+ oneof accelerator_value {
76
+ GPUAccelerator gpu_accelerator = 1;
77
+ }
78
+ }
79
+
64
80
  // Task Metadata
65
81
  message TaskMetadata {
66
82
  // Indicates whether the system should attempt to lookup this task's output to avoid duplication of work.
@@ -106,6 +122,9 @@ message TaskMetadata {
106
122
  // task creates a k8s Pod. If this value is set, the specified PodTemplate will be used instead of, but applied
107
123
  // identically as, the default PodTemplate configured in FlytePropeller.
108
124
  string pod_template_name = 12;
125
+
126
+ // Additional metadata associated with resources to allocate to this task
127
+ ResourceMetadata resource_metadata = 13;
109
128
  }
110
129
 
111
130
  // A Task structure that uniquely identifies a task in the system
@@ -279,4 +279,7 @@ message WorkflowTemplate {
279
279
  message TaskNodeOverrides {
280
280
  // A customizable interface to convey resources requested for a task container.
281
281
  Resources resources = 1;
282
+
283
+ // Additional metadata associated with resources to allocate to this task
284
+ core.ResourceMetadata resource_metadata = 2;
282
285
  }