Skip to content

Commit b5dc676

Browse files
author
awstools
committed
feat(client-sagemaker): Adds support for GB200 UltraServers in Amazon SageMaker training jobs, training plans, and HyperPod clusters
1 parent 794fce3 commit b5dc676

35 files changed

+3298
-1599
lines changed

‎clients/client-sagemaker/README.md

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1690,6 +1690,14 @@ DescribeProject
16901690

16911691
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DescribeProjectCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeProjectCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeProjectCommandOutput/)
16921692

1693+
</details>
1694+
<details>
1695+
<summary>
1696+
DescribeReservedCapacity
1697+
</summary>
1698+
1699+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/DescribeReservedCapacityCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeReservedCapacityCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/DescribeReservedCapacityCommandOutput/)
1700+
16931701
</details>
16941702
<details>
16951703
<summary>
@@ -2506,6 +2514,14 @@ ListTrials
25062514

25072515
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/ListTrialsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListTrialsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListTrialsCommandOutput/)
25082516

2517+
</details>
2518+
<details>
2519+
<summary>
2520+
ListUltraServersByReservedCapacity
2521+
</summary>
2522+
2523+
[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/sagemaker/command/ListUltraServersByReservedCapacityCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListUltraServersByReservedCapacityCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-sagemaker/Interface/ListUltraServersByReservedCapacityCommandOutput/)
2524+
25092525
</details>
25102526
<details>
25112527
<summary>

‎clients/client-sagemaker/src/SageMaker.ts

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -868,6 +868,11 @@ import {
868868
DescribeProjectCommandInput,
869869
DescribeProjectCommandOutput,
870870
} from "./commands/DescribeProjectCommand";
871+
import {
872+
DescribeReservedCapacityCommand,
873+
DescribeReservedCapacityCommandInput,
874+
DescribeReservedCapacityCommandOutput,
875+
} from "./commands/DescribeReservedCapacityCommand";
871876
import {
872877
DescribeSpaceCommand,
873878
DescribeSpaceCommandInput,
@@ -1334,6 +1339,11 @@ import {
13341339
ListTrialComponentsCommandOutput,
13351340
} from "./commands/ListTrialComponentsCommand";
13361341
import { ListTrialsCommand, ListTrialsCommandInput, ListTrialsCommandOutput } from "./commands/ListTrialsCommand";
1342+
import {
1343+
ListUltraServersByReservedCapacityCommand,
1344+
ListUltraServersByReservedCapacityCommandInput,
1345+
ListUltraServersByReservedCapacityCommandOutput,
1346+
} from "./commands/ListUltraServersByReservedCapacityCommand";
13371347
import {
13381348
ListUserProfilesCommand,
13391349
ListUserProfilesCommandInput,
@@ -1898,6 +1908,7 @@ const commands = {
18981908
DescribePipelineExecutionCommand,
18991909
DescribeProcessingJobCommand,
19001910
DescribeProjectCommand,
1911+
DescribeReservedCapacityCommand,
19011912
DescribeSpaceCommand,
19021913
DescribeStudioLifecycleConfigCommand,
19031914
DescribeSubscribedWorkteamCommand,
@@ -2000,6 +2011,7 @@ const commands = {
20002011
ListTransformJobsCommand,
20012012
ListTrialComponentsCommand,
20022013
ListTrialsCommand,
2014+
ListUltraServersByReservedCapacityCommand,
20032015
ListUserProfilesCommand,
20042016
ListWorkforcesCommand,
20052017
ListWorkteamsCommand,
@@ -5012,6 +5024,23 @@ export interface SageMaker {
50125024
cb: (err: any, data?: DescribeProjectCommandOutput) => void
50135025
): void;
50145026

5027+
/**
5028+
* @see {@link DescribeReservedCapacityCommand}
5029+
*/
5030+
describeReservedCapacity(
5031+
args: DescribeReservedCapacityCommandInput,
5032+
options?: __HttpHandlerOptions
5033+
): Promise<DescribeReservedCapacityCommandOutput>;
5034+
describeReservedCapacity(
5035+
args: DescribeReservedCapacityCommandInput,
5036+
cb: (err: any, data?: DescribeReservedCapacityCommandOutput) => void
5037+
): void;
5038+
describeReservedCapacity(
5039+
args: DescribeReservedCapacityCommandInput,
5040+
options: __HttpHandlerOptions,
5041+
cb: (err: any, data?: DescribeReservedCapacityCommandOutput) => void
5042+
): void;
5043+
50155044
/**
50165045
* @see {@link DescribeSpaceCommand}
50175046
*/
@@ -6679,6 +6708,23 @@ export interface SageMaker {
66796708
cb: (err: any, data?: ListTrialsCommandOutput) => void
66806709
): void;
66816710

6711+
/**
6712+
* @see {@link ListUltraServersByReservedCapacityCommand}
6713+
*/
6714+
listUltraServersByReservedCapacity(
6715+
args: ListUltraServersByReservedCapacityCommandInput,
6716+
options?: __HttpHandlerOptions
6717+
): Promise<ListUltraServersByReservedCapacityCommandOutput>;
6718+
listUltraServersByReservedCapacity(
6719+
args: ListUltraServersByReservedCapacityCommandInput,
6720+
cb: (err: any, data?: ListUltraServersByReservedCapacityCommandOutput) => void
6721+
): void;
6722+
listUltraServersByReservedCapacity(
6723+
args: ListUltraServersByReservedCapacityCommandInput,
6724+
options: __HttpHandlerOptions,
6725+
cb: (err: any, data?: ListUltraServersByReservedCapacityCommandOutput) => void
6726+
): void;
6727+
66826728
/**
66836729
* @see {@link ListUserProfilesCommand}
66846730
*/

‎clients/client-sagemaker/src/SageMakerClient.ts

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -542,6 +542,10 @@ import {
542542
DescribeProcessingJobCommandOutput,
543543
} from "./commands/DescribeProcessingJobCommand";
544544
import { DescribeProjectCommandInput, DescribeProjectCommandOutput } from "./commands/DescribeProjectCommand";
545+
import {
546+
DescribeReservedCapacityCommandInput,
547+
DescribeReservedCapacityCommandOutput,
548+
} from "./commands/DescribeReservedCapacityCommand";
545549
import { DescribeSpaceCommandInput, DescribeSpaceCommandOutput } from "./commands/DescribeSpaceCommand";
546550
import {
547551
DescribeStudioLifecycleConfigCommandInput,
@@ -815,6 +819,10 @@ import {
815819
ListTrialComponentsCommandOutput,
816820
} from "./commands/ListTrialComponentsCommand";
817821
import { ListTrialsCommandInput, ListTrialsCommandOutput } from "./commands/ListTrialsCommand";
822+
import {
823+
ListUltraServersByReservedCapacityCommandInput,
824+
ListUltraServersByReservedCapacityCommandOutput,
825+
} from "./commands/ListUltraServersByReservedCapacityCommand";
818826
import { ListUserProfilesCommandInput, ListUserProfilesCommandOutput } from "./commands/ListUserProfilesCommand";
819827
import { ListWorkforcesCommandInput, ListWorkforcesCommandOutput } from "./commands/ListWorkforcesCommand";
820828
import { ListWorkteamsCommandInput, ListWorkteamsCommandOutput } from "./commands/ListWorkteamsCommand";
@@ -1212,6 +1220,7 @@ export type ServiceInputTypes =
12121220
| DescribePipelineExecutionCommandInput
12131221
| DescribeProcessingJobCommandInput
12141222
| DescribeProjectCommandInput
1223+
| DescribeReservedCapacityCommandInput
12151224
| DescribeSpaceCommandInput
12161225
| DescribeStudioLifecycleConfigCommandInput
12171226
| DescribeSubscribedWorkteamCommandInput
@@ -1314,6 +1323,7 @@ export type ServiceInputTypes =
13141323
| ListTransformJobsCommandInput
13151324
| ListTrialComponentsCommandInput
13161325
| ListTrialsCommandInput
1326+
| ListUltraServersByReservedCapacityCommandInput
13171327
| ListUserProfilesCommandInput
13181328
| ListWorkforcesCommandInput
13191329
| ListWorkteamsCommandInput
@@ -1584,6 +1594,7 @@ export type ServiceOutputTypes =
15841594
| DescribePipelineExecutionCommandOutput
15851595
| DescribeProcessingJobCommandOutput
15861596
| DescribeProjectCommandOutput
1597+
| DescribeReservedCapacityCommandOutput
15871598
| DescribeSpaceCommandOutput
15881599
| DescribeStudioLifecycleConfigCommandOutput
15891600
| DescribeSubscribedWorkteamCommandOutput
@@ -1686,6 +1697,7 @@ export type ServiceOutputTypes =
16861697
| ListTransformJobsCommandOutput
16871698
| ListTrialComponentsCommandOutput
16881699
| ListTrialsCommandOutput
1700+
| ListUltraServersByReservedCapacityCommandOutput
16891701
| ListUserProfilesCommandOutput
16901702
| ListWorkforcesCommandOutput
16911703
| ListWorkteamsCommandOutput

‎clients/client-sagemaker/src/commands/AttachClusterNodeVolumeCommand.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ export interface AttachClusterNodeVolumeCommandInput extends AttachClusterNodeVo
2828
export interface AttachClusterNodeVolumeCommandOutput extends AttachClusterNodeVolumeResponse, __MetadataBearer {}
2929

3030
/**
31-
* <p> Attaches your Amazon Elastic Block Store (Amazon EBS) volume to a node in your EKS-orchestrated HyperPod cluster. </p> <p> This API works with the Amazon Elastic Block Store (Amazon EBS) Container Storage Interface (CSI) driver to manage the lifecycle of persistent storage in your HyperPod EKS clusters. </p>
31+
* <p> Attaches your Amazon Elastic Block Store (Amazon EBS) volume to a node in your EKS orchestrated HyperPod cluster. </p> <p> This API works with the Amazon Elastic Block Store (Amazon EBS) Container Storage Interface (CSI) driver to manage the lifecycle of persistent storage in your HyperPod EKS clusters. </p>
3232
* @example
3333
* Use a bare-bones client and the command you need to make an API call.
3434
* ```javascript

0 commit comments

Comments
 (0)