Type-stabilize sn_quadratic_coefficient

tkf · Jul 3, 2018 · 61d61ee · 61d61ee
1 parent 228f4d7
commit 61d61ee
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/codim2/analysis.jl b/src/codim2/analysis.jl
@@ -79,10 +79,12 @@ left_eigvec(tkind, J) = right_eigvec(tkind, J')  # TODO: optimize
 
 function sn_quadratic_coefficient(tkind, ::ImmutableState, wrapper)
     cache = as(wrapper, ContinuationCache)
+    x0 = ds_state(cache)
     J = ds_jacobian(cache)
     q = right_eigvec(tkind, J)
     p = left_eigvec(tkind, J)
-    x0 = ds_state(cache)
+    q = cast_container(typeof(x0), q)
+    p = cast_container(typeof(x0), p)
     second_derivative = ForwardDiff.hessian(
         t -> p ⋅ ds_f((@. t[1] * q + x0), cache),
         SVector(zero(x0)),