Skip to content

Commit 76492e9

Browse files
bdicemeta-codesync[bot]
authored andcommitted
feat(cudf): Disable async MR priming in Velox cuDF (#14997)
Summary: This PR skips the default "priming" step of the RMM async memory resource. This reduces initialization costs and has benefits for multi-process applications. xref: - rapidsai/rmm#2060 - rapidsai/rmm#1931 - rapidsai/rmm#2051 Pull Request resolved: #14997 Reviewed By: Yuhta Differential Revision: D83668107 Pulled By: kgpai fbshipit-source-id: 41b6bd5807f60b0e1c76a1c91dd26f7e5451255a
1 parent 74466b4 commit 76492e9

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

velox/experimental/cudf/exec/Utilities.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,8 @@ namespace {
5050
makeCudaMr(), rmm::percent_of_free_device_memory(percent));
5151
}
5252

53-
[[nodiscard]] auto makeAsyncMr(int percent) {
54-
return std::make_shared<rmm::mr::cuda_async_memory_resource>(
55-
rmm::percent_of_free_device_memory(percent));
53+
[[nodiscard]] auto makeAsyncMr() {
54+
return std::make_shared<rmm::mr::cuda_async_memory_resource>();
5655
}
5756

5857
[[nodiscard]] auto makeManagedMr() {
@@ -78,7 +77,7 @@ std::shared_ptr<rmm::mr::device_memory_resource> createMemoryResource(
7877
if (mode == "pool")
7978
return makePoolMr(percent);
8079
if (mode == "async")
81-
return makeAsyncMr(percent);
80+
return makeAsyncMr();
8281
if (mode == "arena")
8382
return makeArenaMr(percent);
8483
if (mode == "managed")

0 commit comments

Comments
 (0)