All zero scalars case handled

icicle/appUtils/msm/msm.cu

@@ -600,9 +600,9 @@ namespace msm {
       unsigned TOTAL_THREADS = 129000; // TODO: device dependent
       unsigned cutoff_run_length = max(2, h_nof_buckets_to_compute / TOTAL_THREADS);
       unsigned cutoff_nof_runs = (h_nof_buckets_to_compute + cutoff_run_length - 1) / cutoff_run_length;
-      NUM_THREADS = min(1 << 5, cutoff_nof_runs);
+      NUM_THREADS = 1 << 5;
       NUM_BLOCKS = (cutoff_nof_runs + NUM_THREADS - 1) / NUM_THREADS;
-      if (bucket_th > 0)
+      if (h_nof_buckets_to_compute > 0 && bucket_th > 0)
         find_cutoff_kernel<S><<<NUM_BLOCKS, NUM_THREADS, 0, stream>>>(
           sorted_bucket_sizes, h_nof_buckets_to_compute, bucket_th, cutoff_run_length, nof_large_buckets);
       unsigned h_nof_large_buckets;

wrappers/rust/icicle-core/src/msm/tests.rs

@@ -171,13 +171,9 @@ where
     let rng = &mut test_rng();
     for test_size in test_sizes {
         for batch_size in batch_sizes {
-            let mut points = generate_random_affine_points_with_zeroes(test_size * batch_size, 5);
+            let points = generate_random_affine_points_with_zeroes(test_size * batch_size, 100);
             let mut scalars = vec![C::ScalarField::zero(); test_size * batch_size];
 
-            // add some zero points
-            for _ in 0..100 {
-                points[rng.gen_range(0..test_size * batch_size)] = Affine::<C>::zero();
-            }
             for _ in 0..(test_size * batch_size) {
                 scalars[rng.gen_range(0..test_size * batch_size)] = C::ScalarField::one();
             }