Skip to content

Commit 6e5f0b7

Browse files
committed
Format Profiler.jl
1 parent 4b759a7 commit 6e5f0b7

1 file changed

Lines changed: 10 additions & 7 deletions

File tree

src/Profiler.jl

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -131,14 +131,14 @@ function with_profiler(
131131
params_content = read(nvidia_params, String)
132132
if contains(params_content, "RmProfilingAdminOnly: 1")
133133
@warn "CUPTI PM counter collection requires profiling permissions. " *
134-
"Set NVreg_RestrictProfilingToAdminUsers=0 in " *
135-
"/etc/modprobe.d/nvidia-profiler.conf and reload the nvidia module. " *
136-
"Continuing without PM counters."
134+
"Set NVreg_RestrictProfilingToAdminUsers=0 in " *
135+
"/etc/modprobe.d/nvidia-profiler.conf and reload the nvidia module. " *
136+
"Continuing without PM counters."
137137
pm_counters = nothing
138138
end
139139
elseif !Sys.islinux()
140140
@warn "PM counter collection is only supported on Linux with NVIDIA GPUs. " *
141-
"Continuing without PM counters."
141+
"Continuing without PM counters."
142142
pm_counters = nothing
143143
end
144144
if pm_counters !== nothing
@@ -149,10 +149,13 @@ function with_profiler(
149149
config_keys = collect(keys(config))
150150
config_values = collect(values(config))
151151
profiler = GC.@preserve config_keys config_values begin
152-
key_ptrs = isempty(config_keys) ? C_NULL : Base.unsafe_convert.(Cstring, config_keys)
153-
val_ptrs = isempty(config_values) ? C_NULL : Base.unsafe_convert.(Cstring, config_values)
152+
key_ptrs =
153+
isempty(config_keys) ? C_NULL : Base.unsafe_convert.(Cstring, config_keys)
154+
val_ptrs =
155+
isempty(config_values) ? C_NULL : Base.unsafe_convert.(Cstring, config_values)
154156
Reactant.MLIR.API.CreateProfilerSession(
155-
device_tracer_level, host_tracer_level,
157+
device_tracer_level,
158+
host_tracer_level,
156159
isempty(config_keys) ? C_NULL : key_ptrs,
157160
isempty(config_values) ? C_NULL : val_ptrs,
158161
Cint(length(config_keys)),

0 commit comments

Comments
 (0)