From 7b9170f1ab0af40384bb75943e9cae26bccc50d8 Mon Sep 17 00:00:00 2001 From: Julia Sloan Date: Mon, 22 Apr 2024 16:16:40 -0700 Subject: [PATCH] show_mem_usage include PipeEndpoint --- experiments/AMIP/coupler_driver.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiments/AMIP/coupler_driver.jl b/experiments/AMIP/coupler_driver.jl index 36822f2b60..ae74352608 100644 --- 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)