show_mem_usage include PipeEndpoint

CliMA · Apr 22, 2024 · 7b9170f · 7b9170f
1 parent 6cd2dd8
commit 7b9170f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/AMIP/coupler_driver.jl b/experiments/AMIP/coupler_driver.jl
@@ -44,7 +44,7 @@ directly compared.
 `objects`: Dict mapping objects whose memory footprint is displayed in the CPU case to their names
 `io`: The input/output stream to use. Currently, `stdout` and `IOBuffer` types are supported.
 """
-function show_memory_usage(comms_ctx, objects, io::Union{Base.TTY, IOContext, IOBuffer} = stdout)
+function show_memory_usage(comms_ctx, objects, io::Union{Base.TTY, IOContext, IOBuffer, Base.PipeEndpoint} = stdout)
     if comms_ctx.device isa ClimaComms.CUDADevice
         # If `io` is `stdout`, print the memory status, otherwise store in buffer to return
         CUDA.memory_status(io)