fix pydocstyle, doc

illinois-ceesd · Aug 27, 2024 · f3e481c · f3e481c
1 parent 74a5fd0
commit f3e481c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mirgecom/logging_quantities.py b/mirgecom/logging_quantities.py
@@ -29,7 +29,8 @@
 .. autoclass:: DiscretizationBasedQuantity
 .. autoclass:: KernelProfile
 .. autoclass:: PythonMemoryUsage
-.. autoclass:: DeviceMemoryUsage
+.. autoclass:: DeviceMemoryUsageCUDA
+.. autoclass:: DeviceMemoryUsageAMD
 .. autofunction:: initialize_logmgr
 .. autofunction:: logmgr_add_cl_device_info
 .. autofunction:: logmgr_add_device_memory_usage
@@ -468,7 +469,6 @@ def __init__(self, dev: "cl.Device", name: Optional[str] = None) -> None:
 
     def __call__(self) -> Optional[float]:
         """Return the memory usage in MByte."""
-
         # NB: dev.global_mem_size is in Bytes,
         #     dev.global_free_memory_amd is in KByte,
         #     the actual granularity of the returned values appears to be MByte