From 37afcf2e9e38f645258e56697f8a39c7796c1451 Mon Sep 17 00:00:00 2001 From: Elias Bermudez <6505145+debermudez@users.noreply.github.com> Date: Wed, 18 Nov 2020 09:06:37 -0800 Subject: [PATCH] Adjust range of number of ops found (#124) --- qa/L0_lenet/test_lenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/L0_lenet/test_lenet.py b/qa/L0_lenet/test_lenet.py index 3b4e3ae..88de614 100755 --- a/qa/L0_lenet/test_lenet.py +++ b/qa/L0_lenet/test_lenet.py @@ -94,7 +94,7 @@ def test_run_profile(self): # verify final id is in the range # Which kernel cuDNN uses is nondeterministic. # While the exact number of kernels is not clear, for this network, it should be [60, 70] - self.assertTrue(int(row['Idx']) in range(60, 71), f"Final Idx: {row['Idx']}") + self.assertTrue(int(row['Idx']) in range(65, 75), f"Final Idx: {row['Idx']}") if __name__ == '__main__':