apacheGH-41664: PrettyPrint non-cpu data by copying slice to default …

…CPU device
jorisvandenbossche · Jun 6, 2024 · 9944230 · 9944230
1 parent dbcce63
commit 9944230
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cpp/src/arrow/pretty_print.cc b/cpp/src/arrow/pretty_print.cc
@@ -425,6 +425,10 @@ Status PrettyPrint(const Array& arr, int indent, std::ostream* sink) {
 Status PrettyPrint(const Array& arr, const PrettyPrintOptions& options,
                    std::ostream* sink) {
   ArrayPrinter printer(options, sink);
+  if (arr.device_type() != DeviceAllocationType::kCPU) {
+    auto arr_sliced = arr.CopyTo(default_cpu_memory_manager());
+    return printer.Print(arr);
+  }
   return printer.Print(arr);
 }