Skip to content

feature: TF242 ioc support #2982

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 16 commits into from
Mar 18, 2022
Merged

feature: TF242 ioc support #2982

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
ea54793
update TF242 ioc support
Qingzi-Lan Mar 9, 2022
5ca8c57
lint
Qingzi-Lan Mar 9, 2022
7330099
trigger test
Qingzi-Lan Mar 14, 2022
076f6f9
revert trigger test
Qingzi-Lan Mar 14, 2022
ede040a
trigger test
Qingzi-Lan Mar 15, 2022
656f960
revert trigger test
Qingzi-Lan Mar 15, 2022
34af8a8
update test framework version
Qingzi-Lan Mar 16, 2022
223d4de
update
Qingzi-Lan Mar 16, 2022
5657fd4
revert previous change
Qingzi-Lan Mar 16, 2022
fac621a
update
Qingzi-Lan Mar 16, 2022
5affbb4
update
Qingzi-Lan Mar 16, 2022
49dc59f
update
Qingzi-Lan Mar 16, 2022
b9d0e27
format
Qingzi-Lan Mar 16, 2022
565af7c
update
Qingzi-Lan Mar 16, 2022
a496a7a
format
Qingzi-Lan Mar 16, 2022
dd4b5a1
Merge branch 'master' into TF242-IOC-update
mufaddal-rohawala Mar 17, 2022
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
3 changes: 3 additions & 0 deletions src/sagemaker/image_uri_config/neo-tensorflow.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@
"1.12.0": "1.15.3",
"1.13.0": "1.15.3",
"1.14.0": "1.15.3",
"2.4": "2.4.2",
"2.4.0": "2.4.2",
"2.4.1": "2.4.2",
"2.4.2": "2.4.2"
},
"versions": {
Expand Down
1 change: 1 addition & 0 deletions src/sagemaker/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -621,6 +621,7 @@ def _compilation_job_config(

if (
framework.lower() == "pytorch"
or framework.lower() == "tensorflow"
and re.match("(?=^ml_)(?!ml_inf)", target_instance_type) is not None
and framework_version is not None
):
Expand Down