zhztheplayer commented on code in PR #9634:
URL: https://github.com/apache/incubator-gluten/pull/9634#discussion_r2111802643
##########
cpp/core/jni/JniWrapper.cc:
##########
@@ -397,11 +397,15 @@
Java_org_apache_gluten_vectorized_PlanEvaluatorJniWrapper_nativeCreateKernelWith
jint partitionId,
jlong taskId,
jboolean enableDumping,
- jstring spillDir) {
+ jstring spillDir,
+ jboolean enableCudf) {
JNI_METHOD_START
auto ctx = getRuntime(env, wrapper);
- auto& conf = ctx->getConfMap();
+ auto conf = ctx->getConfMap();
+ if (enableCudf) {
+ conf[kCudfEnabled] = "true";
+ }
Review Comment:
Looks good so far. Opened
https://github.com/apache/incubator-gluten/issues/9785 for further enhancement.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
