[JAX SC] Adjust fuzz test ranges for GetColIdIsCorrect.

The input domains for the `GetColIdIsCorrect` fuzz test are updated to better cover relevant ranges. `embedding_id` range is increased, `col_shift` is narrowed, `col_offset_per_shard` is slightly reduced, and `num_scs_bit` is adjusted.

PiperOrigin-RevId: 821830517

Author: adityagupta1089

jax_tpu_embedding/sparsecore/lib/core/input_preprocessing_util_test.cc

@@ -77,10 +77,10 @@ void GetColIdIsCorrect(int embedding_id, int col_shift,
 }
 
 FUZZ_TEST(InputPreprocessingUtilTest, GetColIdIsCorrect)
-    .WithDomains(/*embedding_id=*/fuzztest::InRange(0, 1000000),
-                 /*col_shift=*/fuzztest::InRange(0, 1000000),
-                 /*col_offset_per_shard=*/fuzztest::InRange(0, 1000000),
-                 /*num_scs_bit=*/fuzztest::InRange(0, 10));
+    .WithDomains(/*embedding_id=*/fuzztest::InRange(0, 100000000),
+                 /*col_shift=*/fuzztest::InRange(0, 64),
+                 /*col_offset_per_shard=*/fuzztest::InRange(0, 600000),
+                 /*num_scs_bit=*/fuzztest::InRange(0, 9));
 
 TEST(InputPreprocessingUtilTest, CeilOfRatio) {
   EXPECT_EQ(CeilOfRatio(/*numerator=*/1, /*denominator=*/1), 1);