fix(gpu): panics

Author: andrei-stoian-zama

backends/tfhe-cuda-backend/cuda/src/utils/helper_multi_gpu.cuh

@@ -44,7 +44,7 @@ void multi_gpu_copy_array_from_cpu_async(CudaStreams streams,
                                          Torus const *h_src,
                                          uint32_t elements_per_gpu,
                                          bool gpu_memory_allocated) {
-  PANIC_IF_FALSE(dest.size() == streams.count(),
+  PANIC_IF_FALSE(dest.size() <= streams.count(),
                  "Destination multi-GPU LWE array was allocated on a different "
                  "number of GPUs than requested in this call. ");
   for (uint i = 0; i < streams.count(); i++) {
@@ -158,7 +158,7 @@ void multi_gpu_scatter_lwe_async(CudaStreams streams,
   if (max_active_gpu_count < streams.count())
     PANIC("Cuda error: number of gpus in scatter should be <= number of gpus "
           "used to create the lut")
-  PANIC_IF_FALSE(dest.size() == streams.count(),
+  PANIC_IF_FALSE(dest.size() <= streams.count(),
                  "Destination multi-GPU LWE array was allocated on a different "
                  "number of GPUs than requested in this call. ");
   for (uint i = 0; i < streams.count(); i++) {