-
Notifications
You must be signed in to change notification settings - Fork 82
Test Updates For TF 2.4 #423
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.
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
Conversation
Codecov Report
@@ Coverage Diff @@
## master #423 +/- ##
==========================================
- Coverage 77.70% 75.20% -2.51%
==========================================
Files 113 113
Lines 10143 10143
==========================================
- Hits 7882 7628 -254
- Misses 2261 2515 +254
Continue to review full report at Codecov.
|
ndodda-amazon
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good, left a minor comment.
smdebug/tensorflow/utils.py
Outdated
|
|
||
| def is_profiler_supported_for_tf_version(): | ||
| return is_tf_version_2_2_x() or is_tf_version_2_3_x() | ||
| return is_tf_version_2_2_x() or is_tf_version_2_3_x() or is_tf_version_greater_than_2_4_x() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Two nits:
- Maybe we can just simplify this and do
return version.parse("2.2.0") <= TF_VERSION - Naming schema clashes with
is_tf_version_greater_than_2_4_x, we should just rename this function tois_tf_version_greater_than_2_2_xor renameis_tf_version_greater_than_2_4_xappropriately.
Description of changes:
tf2_4.ymlbuildspec to test the changes on a custom version of tf 2.4.0 gpuis_profiler_supportedfunction to include tf 2.4.0Style and formatting:
I have run
pre-commit installto ensure that auto-formatting happens with every commit.Issue number, if available
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.