Skip to content

Commit

Permalink
Fix issues
Browse files Browse the repository at this point in the history
  • Loading branch information
kingcrimsontianyu committed Oct 30, 2024
1 parent 973b312 commit 52aa6ae
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 6 deletions.
6 changes: 3 additions & 3 deletions cpp/src/io/utilities/config_utils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -56,11 +56,11 @@ void set_up_kvikio()
{
static std::once_flag flag{};
std::call_once(flag, [] {
auto compat_mode = kvikio::detail::getenv_or<bool>("KVIKIO_COMPAT_MODE", true);
kvikio::defaults::compat_mode_reset(compat_mode);

auto nthreads = getenv_or<unsigned int>("KVIKIO_NTHREADS", 4U);
kvikio::defaults::thread_pool_nthreads_reset(nthreads);

auto compat_mode = getenv_or<bool>("KVIKIO_COMPAT_MODE", true);
kvikio::defaults::compat_mode_reset(compat_mode);
});
}
} // namespace cufile_integration
Expand Down
11 changes: 8 additions & 3 deletions docs/cudf/source/user_guide/io/io.md
Original file line number Diff line number Diff line change
Expand Up @@ -97,9 +97,14 @@ This variable also controls the GDS compatibility mode.

There are four valid values for the environment variable:

- "GDS": Enable GDS use; GDS compatibility mode is *off*.
- "ALWAYS": Enable GDS use; GDS compatibility mode is *on*.
- "KVIKIO": Enable GDS through [KvikIO](https://github.com/rapidsai/kvikio).
- "GDS": Enable GDS use. If the cuFile library cannot be properly loaded,
fall back to the GDS compatibility mode.
- "ALWAYS": Enable GDS use. If the cuFile library cannot be properly loaded,
throw an exception.
- "KVIKIO": Enable GDS through [KvikIO](https://github.com/rapidsai/kvikio). Note
that if the environment variable `KVIKIO_COMPAT_MODE` is `ON`, or if KvikIO detects
that the system is not properly configured for GDS, the I/O will fall back to the
GDS compatibility mode.
- "OFF": Completely disable GDS use.

If no value is set, behavior will be the same as the "KVIKIO" option.
Expand Down

0 comments on commit 52aa6ae

Please sign in to comment.