More inplace operations for Enzyme

ptiede · Jul 24, 2024 · 2f7fd64 · 2f7fd64
1 parent ff778a6
commit 2f7fd64
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/instrument/model.jl b/src/instrument/model.jl
@@ -220,7 +220,7 @@ function apply_instrument(vis, J::ObservedInstrumentModel, x)
 end
 
 function apply_instrument(vis, J::ObservedInstrumentModel{<:Union{JonesR, JonesF}}, x)
-    vout = intout(vis)
+    vout = vis#intout(vis)
     _apply_instrument!(baseimage(vout), baseimage(vis), J, (;))
     return vout
 end

diff --git a/src/mrf_image.jl b/src/mrf_image.jl
@@ -34,6 +34,8 @@ end
 
 function _apply_fluctuations(t::VLBIImagePriors.LogRatioTransform, mimg::AbstractArray, δ::AbstractArray)
     @argcheck isapprox(sum(mimg), 1, atol=1e-6) "Mean image must have unit flux when using log-ratio transformations in apply_fluctuations"
-    r = baseimage(mimg).*to_simplex(t, δ)
-    return r./sum(r)
+    r = to_simplex(t, δ)
+    r .= r.*parent(mimg)
+    r .= r./sum(r)
+    return r
 end