saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix flaky atomic iter test (#7649)

Browse source
This commit is contained in:
bddppq 2018-05-17 21:17:29 -07:00 committed by GitHub
parent 5207998fc3
commit bc4feab3e3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
caffe2/python/hypothesis_test.py
View file

@ -1713,9 +1713,10 @@ class TestOperators(hu.HypothesisTestCase):
if num_iters * num_nets > 0:
stats_key = self.ws.blobs[("stats_key")].fetch()
self.assertEqual(b'atomic_iter/stats/iter/num_iter', stats_key[0])
atomic_iter_key = b'atomic_iter/stats/iter/num_iter'
self.assertTrue(atomic_iter_key in stats_key)
stat_val = self.ws.blobs[("stats_val")].fetch()
self.assertEqual(num_iters * num_nets, stat_val[0])
self.assertEqual(num_iters * num_nets, stat_val[list(stats_key).index(atomic_iter_key)])
@given(a=hu.tensor(),