Change ReactantCore so that it doesn't depend on Reactant (#1401)

* ReactantCore depended on Reactant in traced

* fix

* bumpver

* Try fix

* fix

---------

Co-authored-by: William S. Moses <[email protected]>
Co-authored-by: William Moses <[email protected]>

Author: ptiede

Project.toml

@@ -1,7 +1,7 @@
 name = "Reactant"
 uuid = "3c362404-f566-11ee-1572-e11a4b42c853"
 authors = ["William Moses <[email protected]>", "Valentin Churavy <[email protected]>", "Sergio Sánchez Ramírez <[email protected]>", "Paul Berg <[email protected]>", "Avik Pal <[email protected]>", "Mosè Giordano <[email protected]>"]
-version = "0.2.134"
+version = "0.2.135"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -89,7 +89,7 @@ Preferences = "1.4"
 PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
-ReactantCore = "0.1.12"
+ReactantCore = "0.1.15"
 Reactant_jll = "0.0.204"
 ScopedValues = "1.3.0"
 Scratch = "1.2"

lib/ReactantCore/Project.toml

@@ -1,7 +1,7 @@
 name = "ReactantCore"
 uuid = "a3311ec8-5e00-46d5-b541-4f83e724a433"
 authors = ["William Moses <[email protected]>", "Valentin Churavy <[email protected]>", "Sergio Sánchez Ramírez <[email protected]>", "Paul Berg <[email protected]>", "Avik Pal <[email protected]>"]
-version = "0.1.14"
+version = "0.1.15"
 
 [deps]
 ExpressionExplorer = "21656369-7473-754a-2065-74616d696c43"

lib/ReactantCore/src/ReactantCore.jl

@@ -3,7 +3,7 @@ module ReactantCore
 using ExpressionExplorer: ExpressionExplorer
 using MacroTools: MacroTools
 
-export @trace, within_compile, MissingTracedValue
+export @trace, within_compile, MissingTracedValue, promote_to_traced
 
 # Traits
 function is_traced((@nospecialize x::T), seen=Base.IdSet()) where {T}
@@ -34,6 +34,10 @@ const SPECIAL_SYMBOLS = [
     :(:), :nothing, :missing, :Inf, :Inf16, :Inf32, :Inf64, :Base, :Core
 ]
 
+# Function definitions needed to be defined for loop tracing
+# Defined in TracedUtils.jl
+function promote_to_traced end
+
 """
     within_compile()
 
@@ -322,16 +326,13 @@ function trace_for(expr; track_numbers, checkpointing, mincut)
         $bounds_defs
 
         if $(within_compile)()
-            $start_sym = Reactant.TracedUtils.promote_to(
-                Reactant.TracedRNumber{Reactant.unwrapped_eltype(typeof($start_sym))},
+            $start_sym = $(ReactantCore.promote_to_traced)(
                 $start_sym,
             )
-            $limit_sym = Reactant.TracedUtils.promote_to(
-                Reactant.TracedRNumber{Reactant.unwrapped_eltype(typeof($limit_sym))},
+            $limit_sym = $(ReactantCore.promote_to_traced)(
                 $limit_sym,
             )
-            $step_sym = Reactant.TracedUtils.promote_to(
-                Reactant.TracedRNumber{Reactant.unwrapped_eltype(typeof($step_sym))},
+            $step_sym = $(ReactantCore.promote_to_traced)(
                 $step_sym,
             )
         end

src/TracedUtils.jl

@@ -15,9 +15,14 @@ using ..Reactant:
     ReactantPrimitive,
     Ops
 using ReactantCore: ReactantCore
-using ReactantCore: MissingTracedValue, is_traced, materialize_traced_array
+using ReactantCore: MissingTracedValue, is_traced, materialize_traced_array, promote_to_traced
 using Functors: Functors
 
+function ReactantCore.promote_to_traced(x)
+    return promote_to(Reactant.TracedRNumber{Reactant.unwrapped_eltype(typeof(x))}, x)
+end
+
+
 ReactantCore.materialize_traced_array(x::AbstractArray) = x
 
 ReactantCore.materialize_traced_array(x::TracedRArray) = x