Merge pull request #1726 from CliMA/ck/more_nvtx

Add NVTX annotation
CliMA · May 13, 2024 · 76e5cde · 76e5cde
2 parents b1fc161 + 696fb70
commit 76e5cde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/MatrixFields/field_matrix_iterative_solver.jl b/src/MatrixFields/field_matrix_iterative_solver.jl
@@ -83,7 +83,7 @@ A wrapper for [`lazy_preconditioner`](@ref) that turns the lazy `FieldMatrix`
 `P` into a concrete `FieldMatrix` when the `PreconditionerAlgorithm` `P_alg`
 requires a `FieldMatrixSolverAlgorithm` to invert it.
 """
-function lazy_or_concrete_preconditioner(P_alg, P_cache, A)
+NVTX.@annotate function lazy_or_concrete_preconditioner(P_alg, P_cache, A)
     isnothing(P_alg) && return nothing
     lazy_P = lazy_preconditioner(P_alg, A)
     (is_diagonal(P_alg) || !is_lazy(lazy_P)) && return lazy_P