Cap max number of derivatives to 32 on GPUs

Author: dennisYatunin

src/prognostic_equations/implicit/auto_sparse_jacobian.jl

@@ -282,7 +282,8 @@ function jacobian_cache(
         # max_memory is too small to fit any εs, try using one ε per partition.
         # TODO: Replace the fields in I_matrix_partitions with column fields,
         # making memory_for_I_matrix negligible compared to max_memory.
-        n_εs_max = (max_memory - memory_for_I_matrix) ÷ memory_per_ε - 1
+        n_εs_max =
+            min(32, (max_memory - memory_for_I_matrix) ÷ memory_per_ε - 1)
         cld(n_colors, max(n_εs_max, 1))
     end
     n_εs = cld(n_colors, n_partitions)