Add NVTX and annotate orchestrate_diagnostics

Author: nefrathenrici

Project.toml

@@ -10,6 +10,7 @@ ClimaCore = "d414da3d-4745-48bb-8d80-42e94e092884"
 ClimaUtilities = "b3f4f4ca-9299-4f7f-bd9b-81e1242a7513"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 NCDatasets = "85f8d34a-cbdd-5861-8df4-14fed0d494ab"
+NVTX = "5da4648a-3479-48b8-97b9-01cb529c0a1f"
 OrderedCollections = "bac558e1-5e72-5ebc-8fee-abe8a469f55d"
 SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 
@@ -24,9 +25,10 @@ ClimaUtilities = "0.1.22"
 Dates = "1"
 Documenter = "1"
 ExplicitImports = "1.6"
-LazyBroadcast = "1"
 JuliaFormatter = "1"
+LazyBroadcast = "1"
 NCDatasets = "0.14"
+NVTX = "1.0.0"
 OrderedCollections = "1.4"
 Profile = "1"
 ProfileCanvas = "0.1.6"

src/clima_diagnostics.jl

@@ -1,4 +1,5 @@
 import Accessors
+import NVTX
 import SciMLBase
 import ClimaComms
 import ClimaCore: Spaces
@@ -215,7 +216,7 @@ end
 Loop over all the `ScheduledDiagnostics` in `diagnostic_handler` and run compute
 and output according to their schedule functions.
 """
-function orchestrate_diagnostics(
+NVTX.@annotate function orchestrate_diagnostics(
     integrator,
     diagnostic_handler::DiagnosticsHandler,
 )