Skip to content

Commit 76a025c

Browse files
committed
feature: Enable Profiler in China Regions
1 parent a8ab823 commit 76a025c

File tree

6 files changed

+583
-19
lines changed

6 files changed

+583
-19
lines changed

doc/workflows/pipelines/sagemaker.workflow.pipelines.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,11 @@ Execution_variables
5151

5252
.. autoclass:: sagemaker.workflow.execution_variables.ExecutionVariables
5353

54+
Functions
55+
---------
56+
57+
.. autoclass:: sagemaker.workflow.functions.Join
58+
5459
Parameters
5560
----------
5661

src/sagemaker/fw_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
)
5050

5151
DEBUGGER_UNSUPPORTED_REGIONS = ("us-iso-east-1",)
52-
PROFILER_UNSUPPORTED_REGIONS = ("us-iso-east-1", "cn-north-1", "cn-northwest-1")
52+
PROFILER_UNSUPPORTED_REGIONS = ("us-iso-east-1",)
5353

5454
SINGLE_GPU_INSTANCE_TYPES = ("ml.p2.xlarge", "ml.p3.2xlarge")
5555
SM_DATAPARALLEL_SUPPORTED_INSTANCE_TYPES = (
@@ -59,7 +59,7 @@
5959
"local_gpu",
6060
)
6161
SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
62-
"tensorflow": ["2.3.0", "2.3.1", "2.4.1"],
62+
"tensorflow": ["2.3.1", "2.3.2", "2.4.1"],
6363
"pytorch": ["1.6.0", "1.7.1", "1.8.0"],
6464
}
6565
SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]

0 commit comments

Comments
 (0)