diff --git a/mirgecom/logging_quantities.py b/mirgecom/logging_quantities.py index 12574101b..558aab374 100644 --- a/mirgecom/logging_quantities.py +++ b/mirgecom/logging_quantities.py @@ -457,7 +457,7 @@ def __call__(self) -> Optional[float]: class DeviceMemoryUsageAMD(PostLogQuantity): """Logging support for AMD GPU memory usage.""" - def __init__(self, dev: "cl.Device", name: Optional[str] = None) -> None: + def __init__(self, dev: cl.Device, name: Optional[str] = None) -> None: if name is None: name = "memory_usage_gpu"