Skip to content

Conversation

@kanvi-nervana
Copy link
Contributor

@kanvi-nervana kanvi-nervana commented Aug 5, 2019

This PR implements the following new tests to test the bfloat16 conversions

  1. test_fusedbatchnorm_training_nhwc.py
  2. test_fusedbatchnorm_training_nchw.py
  3. test_maxpoolbackprop_nhwc.py
  4. test_maxpoolbackprop_nhwc.py

The following existing tests are also fixed as part of this PR

  1. test_fusedbatchnorm.py
  2. test_l2loss.py
  3. test_maxpoolbackprop.py
  4. test_relugrad.py
  5. test_sparse_softmax_cross_entropy_with_logits.py

Also, made changes to the cmake so the new folder is included.

Copy link
Contributor

@shresthamalik shresthamalik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Requested some changes

ng_outval = sess_ng.run(ng_out[0], feed_dict=feed_dict)

# transpose TF output from NHWC to NCHW for comparison with ngraph output
assert (np.allclose(
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do we care about only the 1st output of the op?

remove fusedbatchnorm_inferece tests since those are not needed for resnet
Copy link
Contributor

@shresthamalik shresthamalik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@sayantan-nervana sayantan-nervana added fully reviewed ready to merge This PR is the next in the queue. labels Aug 8, 2019
@shresthamalik shresthamalik merged commit 478c89e into master Aug 8, 2019
@shresthamalik shresthamalik deleted the kanvi/bfloat16_tests branch August 8, 2019 02:35
gopoka pushed a commit that referenced this pull request Oct 28, 2019
This PR implements the following new tests to test the bfloat16 conversions

test_fusedbatchnorm_training_nhwc.py
test_fusedbatchnorm_training_nchw.py
test_maxpoolbackprop_nhwc.py
test_maxpoolbackprop_nhwc.py
The following existing tests are also fixed as part of this PR

test_fusedbatchnorm.py
test_l2loss.py
test_maxpoolbackprop.py
test_relugrad.py
test_sparse_softmax_cross_entropy_with_logits.py
Also, made changes to the cmake so the new folder is included.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants