Update grackle cooling with last grackle modifications

src/cooling/grackle/cooling.h

@@ -540,22 +540,12 @@ __attribute__((always_inline)) INLINE static gr_float cooling_rate(
 
   /* solve chemistry */
   chemistry_data chemistry_grackle = cooling->chemistry;
-  chemistry_data_storage my_rates = grackle_rates;
-  int error_code = _solve_chemistry(
-      &chemistry_grackle, &my_rates, &units, dt, data.grid_dx, data.grid_rank,
-      data.grid_dimension, data.grid_start, data.grid_end, data.density,
-      data.internal_energy, data.x_velocity, data.y_velocity, data.z_velocity,
-      data.HI_density, data.HII_density, data.HM_density, data.HeI_density,
-      data.HeII_density, data.HeIII_density, data.H2I_density,
-      data.H2II_density, data.DI_density, data.DII_density, data.HDI_density,
-      data.e_density, data.metal_density, data.volumetric_heating_rate,
-      data.specific_heating_rate, data.RT_heating_rate,
-      data.RT_HI_ionization_rate, data.RT_HeI_ionization_rate,
-      data.RT_HeII_ionization_rate, data.RT_H2_dissociation_rate, NULL);
-  if (error_code == 0) error("Error in solve_chemistry.");
-  // if (solve_chemistry(&units, &data, dt) == 0) {
-  //  error("Error in solve_chemistry.");
-  //}
+  chemistry_data_storage chemistry_rates = grackle_rates;
+  if (local_solve_chemistry(&chemistry_grackle,
+			    &chemistry_rates,
+			    &units, &data, dt) == 0) {
+    error("Error in solve_chemistry.");
+  }
 
   /* copy from grackle data to particle */
   cooling_copy_from_grackle(data, p, xp, density);
@@ -623,7 +613,10 @@ __attribute__((always_inline)) INLINE static gr_float cooling_time(
 
   /* Compute cooling time */
   gr_float cooling_time;
-  if (calculate_cooling_time(&units, &data, &cooling_time) == 0) {
+  chemistry_data chemistry_grackle = cooling->chemistry;
+  chemistry_data_storage chemistry_rates = grackle_rates;
+  if (local_calculate_cooling_time(&chemistry_grackle, &chemistry_rates,
+				   &units, &data, &cooling_time) == 0) {
     error("Error in calculate_cooling_time.");
   }