Change print debug

Signed-off-by: Nghia Truong <[email protected]>
rapidsai · Oct 12, 2024 · 53038ca · 53038ca
1 parent 14b5688
commit 53038ca
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/cpp/tests/io/orc_chunked_reader_test.cu b/cpp/tests/io/orc_chunked_reader_test.cu
@@ -130,20 +130,20 @@ auto chunked_read(std::string const& filepath,
     }
     ++num_chunks;
 
-    if (chunk.tbl->num_columns() == 3) {
-      auto col2  = chunk.tbl->get_column(2).view();
-      auto slice = cudf::slice(col2, {0, 5})[0];
-      printf("\ncol2, chunk %d: \n", num_chunks);
-      cudf::test::print(slice);
-    }
-
     out_tables.emplace_back(std::move(chunk.tbl));
   } while (reader.has_next());
 
   if (num_chunks > 1) {
     CUDF_EXPECTS(out_tables.front()->num_rows() != 0, "Number of rows in the new chunk is zero.");
   }
 
+  if (out_tables.front()->num_columns() == 3) {
+    auto col2  = out_tables.front()->get_column(2).view();
+    auto slice = cudf::slice(col2, {0, 5})[0];
+    printf("\ncol2, chunk %d: \n", num_chunks);
+    cudf::test::print(slice);
+  }
+
   auto out_tviews = std::vector<cudf::table_view>{};
   for (auto const& tbl : out_tables) {
     out_tviews.emplace_back(tbl->view());