commit b2277075b0fe5cf085d369a313863f64c6fb50c3
author mantaionut <ionut@janeasystems.com> Mon Jun 26 08:03:27 2023 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Mon Jun 26 08:03:27 2023 +0000
tree e470b7ff3fbe6a5c59dc5e6160d689b0a29a9d72
parent 3c34a00d1b94aa4ceeede498ab878f1c5a25afb1

Fixed benchmark_utils.Fuzzer (#101553)

Use np.randint with int64 since int32 is default on Windows.
change default seed to be between [0, 2**32-1] because that is np.random.RandomState required input

Fixes #51205

Pull Request resolved: https://github.com/pytorch/pytorch/pull/101553
Approved by: https://github.com/kit1980

torch/utils/benchmark/utils/fuzzer.py

diff --git a/torch/utils/benchmark/utils/fuzzer.py b/torch/utils/benchmark/utils/fuzzer.py
index 421219a..7d1ee8e 100644
--- a/torch/utils/benchmark/utils/fuzzer.py
+++ b/torch/utils/benchmark/utils/fuzzer.py
@@ -369,7 +369,7 @@
                 ops will create reproducible Tensors.
         """
         if seed is None:
-            seed = np.random.RandomState().randint(0, 2**63)
+            seed = np.random.RandomState().randint(0, 2 ** 32 - 1, dtype=np.int64)
         self._seed = seed
         self._parameters = Fuzzer._unpack(parameters, FuzzedParameter)
         self._tensors = Fuzzer._unpack(tensors, FuzzedTensor)
@@ -392,7 +392,7 @@
 
     def take(self, n):
         state = np.random.RandomState(self._seed)
-        torch.manual_seed(state.randint(low=0, high=2 ** 63))
+        torch.manual_seed(state.randint(low=0, high=2 ** 63, dtype=np.int64))
         for _ in range(n):
             params = self._generate(state)
             tensors = {}