Skip to content

Conversation

@NihalHarish
Copy link
Contributor

@NihalHarish NihalHarish commented Jan 14, 2021

Description of changes:

  • Added a tf2_4.yml buildspec to test the changes on a custom version of tf 2.4.0 gpu
  • TF 2.4 returns 16 tenors for the BIAS collection
  • Expanded the is_profiler_supported function to include tf 2.4.0

Style and formatting:

I have run pre-commit install to 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.

@codecov-io
Copy link

codecov-io commented Jan 14, 2021

Codecov Report

Merging #423 (801cdbd) into master (e431609) will decrease coverage by 2.50%.
The diff coverage is 100.00%.

Impacted file tree graph

@@            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     
Impacted Files Coverage Δ
smdebug/tensorflow/utils.py 64.28% <100.00%> (-23.81%) ⬇️
...debug/profiler/analysis/notebook_utils/__init__.py 0.00% <0.00%> (-100.00%) ⬇️
smdebug/rules/rule_invoker.py 18.18% <0.00%> (-27.28%) ⬇️
smdebug/tensorflow/callable_cache.py 52.17% <0.00%> (-26.09%) ⬇️
smdebug/tensorflow/singleton_utils.py 83.33% <0.00%> (-16.67%) ⬇️
smdebug/profiler/tf_profiler_parser.py 54.54% <0.00%> (-11.58%) ⬇️
smdebug/tensorflow/collection.py 84.53% <0.00%> (-11.35%) ⬇️
smdebug/tensorflow/keras.py 78.82% <0.00%> (-11.27%) ⬇️
smdebug/rules/action/stop_training_action.py 56.45% <0.00%> (-9.68%) ⬇️
... and 18 more

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update e431609...801cdbd. Read the comment docs.

Copy link
Contributor

@ndodda-amazon ndodda-amazon left a 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.


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()
Copy link
Contributor

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 to is_tf_version_greater_than_2_2_x or rename is_tf_version_greater_than_2_4_x appropriately.

@NihalHarish NihalHarish changed the title Account for More Tensors Captured in TF 2.4 Test Updates For TF 2.4 Jan 15, 2021
@NihalHarish NihalHarish merged commit 99282cd into master Jan 17, 2021
@NihalHarish NihalHarish deleted the update_tests_for_tf24 branch January 17, 2021 02:44
NihalHarish added a commit that referenced this pull request Jan 25, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants