add limits for too large problems

Author: 0xSooki

src/kernels.cc.cu

@@ -268,6 +268,15 @@ cudaError_t calculatePermanent(cudaStream_t stream,
       idx_max *= h_n_ary_limits[i];
     }
 
+    const uint64_t MAX_IDX_MAX = 100000000;
+    if (idx_max > MAX_IDX_MAX)
+    {
+      cudaFree(d_new_matrix);
+      cudaFree(d_new_rows);
+      cudaFree(d_n_ary_limits);
+      throw std::runtime_error("Problem too large: idx_max exceeds safe limit.");
+    }
+
     cuda_err = cudaMalloc(&d_n_ary_limits, n_ary_size * sizeof(int));
     if (cuda_err != cudaSuccess)
     {

src/permanent.hpp

@@ -126,6 +126,12 @@ std::complex<double> permanent(Matrix<std::complex<double>> &A,
     idx_max *= n_ary_limits[i];
   }
 
+  const uint64_t MAX_IDX_MAX = 100000000;
+  if (idx_max > MAX_IDX_MAX)
+  {
+    throw std::runtime_error("Problem too large: idx_max exceeds safe limit.");
+  }
+
   // determine the concurrency of the calculation
   unsigned int n_threads = std::thread::hardware_concurrency();
   int64_t concurrency = static_cast<int64_t>(n_threads) * 4;