skip dynamic shape tests, adjust cpu rng test

ngimel · ngimel · commit 28c48ec431ce · 2024-12-26T11:26:35.000-08:00
diff --git a/aten/src/ATen/test/rng_test.h b/aten/src/ATen/test/rng_test.h
@@ -137,7 +137,9 @@ void test_random_from_to(const at::Device& device) {
               range = static_cast<uint64_t>(max_to) - static_cast<uint64_t>(from) + 1;
               from_case_covered = true;
             }
-            if (range < (1ULL << 32)) {
+            // this is leaking details of implementation into test
+            // we are starting to use random64() at 2^25 to minimize skew due to %
+            if (range < (1ULL << 25)) {
               exp = static_cast<T>(static_cast<int64_t>((static_cast<uint32_t>(val) % range + from)));
             } else {
               exp = static_cast<T>(static_cast<int64_t>((val % range + from)));
diff --git a/test/inductor/test_torchinductor_codegen_dynamic_shapes.py b/test/inductor/test_torchinductor_codegen_dynamic_shapes.py
@@ -231,7 +231,7 @@ def run(*ex, **kwargs):
     "test_pointwise_laguerre_polynomial_l_dynamic_shapes": TestFailure(("cuda", "xpu")),
     "test_pointwise_legendre_polynomial_p_dynamic_shapes": TestFailure(("cuda", "xpu")),
     "test_polar_dynamic_shapes": TestFailure(("cpu", "cuda", "xpu"), is_skip=True),
-    "test_randint_distribution": TestFailure(("cuda",)),
+    "test_randint_distribution_dynamic_shapes": TestFailure(("cuda",)),
     "test_randn_generator_dynamic_shapes": TestFailure(("cpu",)),
     "test_randn_like_empty_dynamic_shapes": TestFailure(("cpu", "cuda", "xpu")),
     "test_single_elem_dynamic_shapes": TestFailure(("cpu",)),
diff --git a/test/inductor/test_torchinductor_dynamic_shapes.py b/test/inductor/test_torchinductor_dynamic_shapes.py
@@ -59,7 +59,7 @@
         ("cpu", "cuda", "xpu")
     ),
     "test_conv_inference_heuristics_dynamic_shapes": TestFailure(("cuda", "xpu")),
-    "test_randint_distribution": TestFailure(("cuda",)),
+    "test_randint_distribution_dynamic_shapes": TestFailure(("cuda",)),
 }
 
 if TEST_WITH_ROCM:

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@`
`59`	`59`	`("cpu", "cuda", "xpu")`
`60`	`60`	`),`
`61`	`61`	`"test_conv_inference_heuristics_dynamic_shapes": TestFailure(("cuda", "xpu")),`
`62`		`- "test_randint_distribution": TestFailure(("cuda",)),`
	`62`	`+ "test_randint_distribution_dynamic_shapes": TestFailure(("cuda",)),`
`63`	`63`	`}`
`64`	`64`
`65`	`65`	`if TEST_WITH_ROCM:`