diff --git a/src/eos/equation_of_state.cpp b/src/eos/equation_of_state.cpp index 8da4c514..ac693a69 100644 --- a/src/eos/equation_of_state.cpp +++ b/src/eos/equation_of_state.cpp @@ -174,9 +174,11 @@ void EquationOfStateImpl::apply_primitive_limiter_(torch::Tensor const& prim) { prim.masked_fill_(torch::isnan(prim), 0.); prim[IDN].clamp_min_(options->density_floor()); - int ny = options->thermo()->vapor_ids().size() + - options->thermo()->cloud_ids().size() - 1; - prim.narrow(0, ICY, ny).clamp_min_(0.); + if (options->thermo()) { + int ny = options->thermo()->vapor_ids().size() + + options->thermo()->cloud_ids().size() - 1; + prim.narrow(0, ICY, ny).clamp_min_(0.); + } prim[IPR].clamp_min_(options->pressure_floor()); }