Skip to content

Commit

Permalink
Update calls to KvikIO's config setter (#18144)
Browse files Browse the repository at this point in the history
## Description
KvikIO has changed the function names of the config setters to improve
clarity (rapidsai/kvikio#644). This PR updates
the setter calls in cuDF accordingly.

## Checklist
- [x] I am familiar with the [Contributing
Guidelines](https://github.com/rapidsai/cudf/blob/HEAD/CONTRIBUTING.md).
- [x] New or existing tests cover these changes.
- [x] The documentation is up to date with these changes.
  • Loading branch information
kingcrimsontianyu authored Mar 3, 2025
1 parent fbbd69d commit 0cf6698
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cpp/src/io/utilities/config_utils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -36,10 +36,10 @@ void set_up_kvikio()
cudaFree(nullptr);

auto const compat_mode = kvikio::getenv_or("KVIKIO_COMPAT_MODE", kvikio::CompatMode::ON);
kvikio::defaults::compat_mode_reset(compat_mode);
kvikio::defaults::set_compat_mode(compat_mode);

auto const nthreads = getenv_or<unsigned int>("KVIKIO_NTHREADS", 4u);
kvikio::defaults::thread_pool_nthreads_reset(nthreads);
kvikio::defaults::set_thread_pool_nthreads(nthreads);
});
}

Expand Down

0 comments on commit 0cf6698

Please sign in to comment.