Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions java/src/main/native/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,18 @@ if(USE_GDS)
endif()
endif()

# Set RMM logging level
set(RMM_LOGGING_LEVEL

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@vyasr Is this the correct way to use rapids-logger? I discussed a bit with @abellina offline but I don't know the proper way to solve this problem.

@bdice bdice Jul 18, 2025

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

There's a related conversation here about what should be precompiled in RMM here: rapidsai/rmm#1983 (comment)

It feels like these two topics might be solvable in similar ways.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Assuming rmm is still being from source as part of the build, this will work, yes. It won't have any effect if you are using precompiled binaries, but since Spark doesn't do that this seems fine. The actual logs observed at runtime may still require setting the runtime log level if the logging functions are used.

"INFO"
CACHE STRING "Choose the logging level."
)
set_property(
CACHE RMM_LOGGING_LEVEL PROPERTY STRINGS "TRACE" "DEBUG" "INFO" "WARN" "ERROR" "CRITICAL" "OFF"
)
message(VERBOSE "CUDF_JNI: RMM_LOGGING_LEVEL = '${RMM_LOGGING_LEVEL}'.")

add_compile_definitions("RMM_LOG_ACTIVE_LEVEL=RAPIDS_LOGGER_LOG_LEVEL_${RMM_LOGGING_LEVEL}")

# ##################################################################################################
# * library targets -------------------------------------------------------------------------------

Expand Down