diff --git a/java/src/main/native/src/CudfJni.cpp b/java/src/main/native/src/CudfJni.cpp index 0c560833bb1..3adcf66579e 100644 --- a/java/src/main/native/src/CudfJni.cpp +++ b/java/src/main/native/src/CudfJni.cpp @@ -78,10 +78,10 @@ static void release_contiguous_table_jni(JNIEnv *env) { } } -jobject contiguous_table_from(JNIEnv *env, cudf::contiguous_split_result &split) { - jlong address = reinterpret_cast(split.all_data->data()); - jlong size = static_cast(split.all_data->size()); - jlong buff_address = reinterpret_cast(split.all_data.get()); +jobject contiguous_table_from(JNIEnv *env, cudf::packed_table &split) { + jlong address = reinterpret_cast(split.data.gpu_data->data()); + jlong size = static_cast(split.data.gpu_data->size()); + jlong buff_address = reinterpret_cast(split.data.gpu_data.get()); int num_columns = split.table.num_columns(); cudf::jni::native_jlongArray views(env, num_columns); for (int i = 0; i < num_columns; i++) { @@ -103,7 +103,7 @@ jobject contiguous_table_from(JNIEnv *env, cudf::contiguous_split_result &split) views.get_jArray(), address, size, buff_address); if (ret != nullptr) { - split.all_data.release(); + split.data.gpu_data.release(); } return ret; } diff --git a/java/src/main/native/src/TableJni.cpp b/java/src/main/native/src/TableJni.cpp index 20afe12baf9..8b607486c8e 100644 --- a/java/src/main/native/src/TableJni.cpp +++ b/java/src/main/native/src/TableJni.cpp @@ -1817,7 +1817,7 @@ JNIEXPORT jobjectArray JNICALL Java_ai_rapids_cudf_Table_contiguousSplit(JNIEnv std::vector indices(n_split_indices.data(), n_split_indices.data() + n_split_indices.size()); - std::vector result = cudf::contiguous_split(*n_table, indices); + std::vector result = cudf::contiguous_split(*n_table, indices); cudf::jni::native_jobjectArray n_result = cudf::jni::contiguous_table_array(env, result.size()); for (int i = 0; i < result.size(); i++) { diff --git a/java/src/main/native/src/cudf_jni_apis.hpp b/java/src/main/native/src/cudf_jni_apis.hpp index a7d955b2bbf..f2e0900e030 100644 --- a/java/src/main/native/src/cudf_jni_apis.hpp +++ b/java/src/main/native/src/cudf_jni_apis.hpp @@ -24,7 +24,7 @@ namespace cudf { namespace jni { -jobject contiguous_table_from(JNIEnv *env, cudf::contiguous_split_result &split); +jobject contiguous_table_from(JNIEnv *env, cudf::packed_table &split); native_jobjectArray contiguous_table_array(JNIEnv *env, jsize length);