Reduce input data size for syncbn unit test by 8x (#28571)

this is a preemptive measure, the unit test was failing occasionally with
cudnn internal error
musl/fix_failed_unittests_in_musl
Yang Zhang 5 years ago committed by GitHub
parent c545b9b673
commit c47bfe98cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -50,9 +50,9 @@ class TestSyncBatchNormOpTraining(unittest.TestCase):
"""Setup."""
#self.dtype = np.float32
self.dtype = np.float64
self.N = 32
self.N = 8
self.C = 16
self.H = 64
self.H = 32
self.W = 32
self.dshape = [self.N, self.C, self.H, self.W]
self.atol = 1e-3
@ -196,9 +196,9 @@ class TestFP16SyncBatchNormOpTraining(TestSyncBatchNormOpTraining):
def setUp(self):
"""Setup."""
self.dtype = np.float16
self.N = 32
self.N = 8
self.C = 16
self.H = 64
self.H = 32
self.W = 32
self.dshape = [self.N, self.C, self.H, self.W]
self.atol = 1e-2

Loading…
Cancel
Save