add profile case

Oneflow-Inc · Jul 28, 2023 · e0899c4 · e0899c4
1 parent 18bcdc4
commit e0899c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/oneflow/test/modules/test_bincount.py b/python/oneflow/test/modules/test_bincount.py
@@ -56,7 +56,7 @@ def test_bincount_0element(test_case):
     @profile(torch.bincount)
     def profile_bincount(test_case):
         torch.bincount(torch.ones(4096).int())
-        # torch.bincount(torch.ones(65536).int())
+        torch.bincount(torch.ones(65536).int())
 
 
 if __name__ == "__main__":