Adjust range of number of ops found (#124)

NVIDIA · Nov 18, 2020 · 37afcf2 · 37afcf2
1 parent 95dc3db
commit 37afcf2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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__':