Skip to content

Commit

Permalink
slightly nicer manifolds in ccw
Browse files Browse the repository at this point in the history
  • Loading branch information
Affie committed Jul 7, 2021
1 parent 03a5773 commit 9718395
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
3 changes: 2 additions & 1 deletion src/FactorGraph.jl
Original file line number Diff line number Diff line change
Expand Up @@ -655,7 +655,8 @@ function prepgenericconvolution(Xi::Vector{<:DFGVariable},
nullhypo=nullhypo,
threadmodel=threadmodel,
inflation=inflation,
partialDims=partialDims
partialDims=partialDims,
manifolds = tuple(getManifold.(Xi)...)
)
#
return ccw
Expand Down
6 changes: 4 additions & 2 deletions src/FactorGraphTypes.jl
Original file line number Diff line number Diff line change
Expand Up @@ -277,7 +277,9 @@ function CommonConvWrapper( fnc::T,
perturb=zeros(zDim),
res::AbstractVector{<:Real}=zeros(zDim),
threadmodel::Type{<:_AbstractThreadModel}=MultiThreaded,
inflation::Real=3.0 ) where {T<:FunctorInferenceType,P,H}
inflation::Real=3.0,
manifolds=tuple(getManifold.(factormetadata.fullvariables)...)
) where {T<:FunctorInferenceType,P,H}
#
return CommonConvWrapper(fnc,
xDim,
Expand All @@ -296,7 +298,7 @@ function CommonConvWrapper( fnc::T,
perturb=perturb, res=res )).(1:Threads.nthreads()),
inflation,
partialDims, # SVector(Int32.()...)
tuple(getManifold.(factormetadata.fullvariables)...))
manifolds,)
end


Expand Down

0 comments on commit 9718395

Please sign in to comment.