Skip to content

Commit fedb836

Browse files
bit more doc
1 parent 6bdb85b commit fedb836

2 files changed

Lines changed: 6 additions & 7 deletions

File tree

arraycontext/impl/pytato/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -351,6 +351,8 @@ def __init__(
351351
# {{{ Profiling functionality
352352

353353
def _wait_and_transfer_profile_events(self) -> None:
354+
"""Wait for all profiling events to finish and transfer the results
355+
to *self._profile_results*."""
354356
import pyopencl as cl
355357
# First, wait for completion of all events
356358
if self._profile_events:

arraycontext/impl/pytato/utils.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
.. autofunction:: transfer_to_numpy
77
88
9-
Profiling-related functions:
10-
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
9+
Profiling-related functions
10+
^^^^^^^^^^^^^^^^^^^^^^^^^^^
1111
1212
.. autofunction:: tabulate_profiling_data
1313
"""
@@ -242,7 +242,7 @@ def tabulate_profiling_data(actx: PytatoPyOpenCLArrayContext) -> pytools.Table:
242242
tbl.add_row(("Kernel", "# Calls", "Time_sum [ns]", "Time_avg [ns]"))
243243

244244
# Precision of results
245-
g = ".4g"
245+
g = ".5g"
246246

247247
total_calls = 0
248248
total_time = 0.0
@@ -256,10 +256,7 @@ def tabulate_profiling_data(actx: PytatoPyOpenCLArrayContext) -> pytools.Table:
256256
if t_sum is not None:
257257
total_time += t_sum
258258

259-
time_sum = f"{t_sum:{g}}"
260-
time_avg = f"{t_avg:{g}}"
261-
262-
tbl.add_row((kernel_name, num_calls, time_sum, time_avg))
259+
tbl.add_row((kernel_name, num_calls, f"{t_sum:{g}}", f"{t_avg:{g}}"))
263260

264261
tbl.add_row(("", "", "", ""))
265262
tbl.add_row(("Total", total_calls, f"{total_time:{g}}", "--"))

0 commit comments

Comments
 (0)