From de0dfb6c8202d6635660f687a3cbd95772993994 Mon Sep 17 00:00:00 2001
From: William Thompson <wthompson@uvic.ca>
Date: Mon, 12 Feb 2024 12:03:13 -0800
Subject: [PATCH] Use threads for RV fitting. Should become a flag.

---
 OctofitterRadialVelocity/src/OctofitterRadialVelocity.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/OctofitterRadialVelocity/src/OctofitterRadialVelocity.jl b/OctofitterRadialVelocity/src/OctofitterRadialVelocity.jl
index d50ae9d4..159ec906 100644
--- a/OctofitterRadialVelocity/src/OctofitterRadialVelocity.jl
+++ b/OctofitterRadialVelocity/src/OctofitterRadialVelocity.jl
@@ -136,7 +136,7 @@ function Octofitter.ln_like(
             orbit = elements[planet_i]
             # Need to structarrays orbit if we want this to SIMD
             planet_mass = θ_system.planets[planet_i].mass
-            for epoch_i in eachindex(epochs)
+            Threads.@threads for epoch_i in eachindex(epochs)
                 rv_star_buf[epoch_i] -= radvel(orbit, epochs[epoch_i], planet_mass*Octofitter.mjup2msol)
             end
             # Zygote version:
@@ -253,6 +253,7 @@ include("harps_rvbank.jl")
 include("harps_dr1.jl")
 include("hires.jl")
 include("lick.jl")
+include("ces.jl")
 include("radvel.jl")