Add MPART_INITIALIZE option (#24)

* Update MParT.jl * Update MParT.jl
MeasureTransport · Jul 8, 2024 · 9a59088 · 9a59088
1 parent 45b7735
commit 9a59088
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/MParT.jl b/src/MParT.jl
@@ -19,13 +19,19 @@ end
 function __init__()
     @initcxx
     threads = get(ENV, "KOKKOS_NUM_THREADS", nothing)
+	init_mpart = true
+    try
+		init_mpart = parse(Bool, lowercase(get(ENV, "MPART_INITIALIZE", "true")))
+    catch e
+		@warn "Expected MPART_INITIALIZE to be a `true`/`false` value. Defaulting to true."
+	end
     opts = StdVector{StdString}()
     if !isnothing(threads)
         push!(opts, StdString("kokkos_num_threads"))
         push!(opts, StdString(string(threads)))
     end
     length(opts) > 0 && @info "Using MParT options: "*join(string.(opts),", ")
-    Initialize(opts)
+    init_mpart && Initialize(opts)
 end
 
 """