Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions sample/sagemaker/2017-07-24/service-2.json
Original file line number Diff line number Diff line change
Expand Up @@ -8327,6 +8327,15 @@
"VolumeSizeInGB":{
"shape":"ClusterEbsVolumeSizeInGB",
"documentation":"<p>The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to <code>/opt/sagemaker</code>.</p>"
},
"VolumeKmsKeyId":{
"shape":"KmsKeyId",
"documentation":"<p>The ID of a KMS key to encrypt the Amazon EBS volume.</p>"
},
"RootVolume":{
"shape":"Boolean",
"documentation":"<p>Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two <code>ClusterEbsVolumeConfig</code> fields to configure both the root and secondary volumes. Set the value to <code>True</code> if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When <code>True</code>:</p> <ul> <li> <p>The configuration is applied to the root volume.</p> </li> <li> <p>You can't specify the <code>VolumeSizeInGB</code> field. The size of the root volume is determined for you.</p> </li> <li> <p>You must specify a KMS key ID for <code>VolumeKmsKeyId</code> to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.</p> </li> </ul> <p>Otherwise, by default, the value is <code>False</code>, and the following applies:</p> <ul> <li> <p>The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.</p> </li> <li> <p>You must specify the <code>VolumeSizeInGB</code> field.</p> </li> <li> <p>You can optionally specify the <code>VolumeKmsKeyId</code> to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.</p> </li> </ul>",
"box":true
}
},
"documentation":"<p>Defines the configuration for attaching an additional Amazon Elastic Block Store (EBS) volume to each instance of the SageMaker HyperPod cluster instance group. To learn more, see <a href=\"https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-hyperpod-release-notes.html#sagemaker-hyperpod-release-notes-20240620\">SageMaker HyperPod release notes: June 20, 2024</a>.</p>"
Expand Down Expand Up @@ -8674,6 +8683,7 @@
"ml.p4d.24xlarge",
"ml.p4de.24xlarge",
"ml.p5.48xlarge",
"ml.p6e-gb200.36xlarge",
"ml.trn1.32xlarge",
"ml.trn1n.32xlarge",
"ml.g5.xlarge",
Expand Down
4 changes: 3 additions & 1 deletion src/sagemaker_core/main/code_injection/shape_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -1447,7 +1447,9 @@
},
"ClusterEbsVolumeConfig": {
"members": [
{"name": "VolumeSizeInGB", "shape": "ClusterEbsVolumeSizeInGB", "type": "integer"}
{"name": "VolumeSizeInGB", "shape": "ClusterEbsVolumeSizeInGB", "type": "integer"},
{"name": "VolumeKmsKeyId", "shape": "KmsKeyId", "type": "string"},
{"name": "RootVolume", "shape": "Boolean", "type": "boolean"},
],
"type": "structure",
},
Expand Down
4 changes: 4 additions & 0 deletions src/sagemaker_core/main/shapes.py
Original file line number Diff line number Diff line change
Expand Up @@ -3414,9 +3414,13 @@ class ClusterEbsVolumeConfig(Base):
Attributes
----------------------
volume_size_in_gb: The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.
volume_kms_key_id: The ID of a KMS key to encrypt the Amazon EBS volume.
root_volume: Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True: The configuration is applied to the root volume. You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you. You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key. Otherwise, by default, the value is False, and the following applies: The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key. You must specify the VolumeSizeInGB field. You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.
"""

volume_size_in_gb: Optional[int] = Unassigned()
volume_kms_key_id: Optional[str] = Unassigned()
root_volume: Optional[bool] = Unassigned()


class ClusterMetadata(Base):
Expand Down
Loading