Add debug code

Signed-off-by: Nghia Truong <[email protected]>
NVIDIA · Nov 27, 2024 · a63179e · a63179e
1 parent aad7900
commit a63179e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/main/cpp/src/test_host_udf_agg.cu b/src/main/cpp/src/test_host_udf_agg.cu
@@ -193,6 +193,21 @@ struct test_udf_simple_type : cudf::host_udf_base {
                                  static_cast<OutputType>(init_value),
                                  thrust::plus<>{});
 
+      if constexpr (std::is_same_v<InputType, double> && std::is_same_v<OutputType, double>) {
+        std::vector<double> in(values.size());
+        CUDF_CUDA_TRY(cudaMemcpyAsync(in.data(),
+                                      values.begin<double>(),
+                                      values.size() * sizeof(double),
+                                      cudaMemcpyDefault,
+                                      stream.value()));
+        stream.synchronize();
+        std::cout << "input: " << std::endl;
+        for (auto x : in) {
+          std::cout << x << ", ";
+        }
+        std::cout << "\nresult: " << result << std::endl;
+      }
+
       auto output = cudf::make_numeric_scalar(output_dtype, stream, mr);
       static_cast<cudf::scalar_type_t<OutputType>*>(output.get())->set_value(result, stream);
       return output;