File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44 */
55#pragma once
66
7+ #include < raft/core/memory_stats_resources.hpp>
78#include < raft/core/resource/device_memory_resource.hpp>
89#include < raft/core/resource/dry_run_flag.hpp>
910#include < raft/core/resource/managed_memory_resource.hpp>
1213#include < raft/mr/dry_run_resource.hpp>
1314#include < raft/mr/host_device_resource.hpp>
1415#include < raft/mr/host_memory_resource.hpp>
15- #include < raft/util/memory_stats_resources.hpp>
1616
1717#include < rmm/cuda_stream_view.hpp>
1818#include < rmm/mr/per_device_resource.hpp>
Load diff This file was deleted.
Original file line number Diff line number Diff line change 66#include " ../test_utils.cuh"
77
88#include < raft/core/device_mdarray.hpp>
9+ #include < raft/core/dry_run_resources.hpp>
910#include < raft/core/resource/cuda_stream.hpp>
1011#include < raft/core/resources.hpp>
1112#include < raft/linalg/rsvd.cuh>
1213#include < raft/random/rng.cuh>
1314#include < raft/util/cuda_utils.cuh>
1415#include < raft/util/cudart_utils.hpp>
15- #include < raft/util/dry_run_resources.hpp>
1616
1717#include < gtest/gtest.h>
1818
Original file line number Diff line number Diff line change 66#include " ../test_utils.cuh"
77
88#include < raft/core/device_mdarray.hpp>
9+ #include < raft/core/dry_run_resources.hpp>
910#include < raft/core/resource/cuda_stream.hpp>
1011#include < raft/linalg/init.cuh>
1112#include < raft/linalg/svd.cuh>
1213#include < raft/util/cuda_utils.cuh>
1314#include < raft/util/cudart_utils.hpp>
14- #include < raft/util/dry_run_resources.hpp>
1515
1616#include < gtest/gtest.h>
1717
Original file line number Diff line number Diff line change 66#include " ../test_utils.cuh"
77
88#include < raft/core/device_mdarray.hpp>
9+ #include < raft/core/dry_run_resources.hpp>
910#include < raft/core/host_mdarray.hpp>
1011#include < raft/core/mdspan.hpp>
1112#include < raft/core/resource/cuda_stream.hpp>
1415#include < raft/matrix/sample_rows.cuh>
1516#include < raft/random/rng.cuh>
1617#include < raft/util/cudart_utils.hpp>
17- #include < raft/util/dry_run_resources.hpp>
1818#include < raft/util/itertools.hpp>
1919
2020#include < gtest/gtest.h>
Original file line number Diff line number Diff line change 77
88#include " test_utils.h"
99
10+ #include < raft/core/dry_run_resources.hpp>
11+ #include < raft/core/memory_stats_resources.hpp>
1012#include < raft/core/resource/cuda_stream.hpp>
1113#include < raft/core/resources.hpp>
1214#include < raft/random/rng.cuh>
1315#include < raft/util/cuda_utils.cuh>
1416#include < raft/util/cudart_utils.hpp>
15- #include < raft/util/dry_run_resources.hpp>
16- #include < raft/util/memory_stats_resources.hpp>
1717
1818#include < rmm/exec_policy.hpp>
1919#include < rmm/mr/per_device_resource.hpp>
Original file line number Diff line number Diff line change 66#include < raft/core/copy.hpp>
77#include < raft/core/device_mdarray.hpp>
88#include < raft/core/device_mdspan.hpp>
9+ #include < raft/core/dry_run_resources.hpp>
910#include < raft/core/host_mdspan.hpp>
1011#include < raft/core/operators.hpp>
1112#include < raft/core/resource/cuda_stream.hpp>
1819#include < raft/random/rng.cuh>
1920#include < raft/stats/mean.cuh>
2021#include < raft/util/cudart_utils.hpp>
21- #include < raft/util/dry_run_resources.hpp>
2222
2323#include < rmm/device_uvector.hpp>
2424
Original file line number Diff line number Diff line change 33 * SPDX-License-Identifier: Apache-2.0
44 */
55
6+ #include < raft/core/dry_run_resources.hpp>
67#include < raft/core/resource/device_memory_resource.hpp>
78#include < raft/core/resource/dry_run_flag.hpp>
89#include < raft/core/resource/managed_memory_resource.hpp>
1213#include < raft/mr/host_device_resource.hpp>
1314#include < raft/mr/host_memory_resource.hpp>
1415#include < raft/util/cudart_utils.hpp>
15- #include < raft/util/dry_run_resources.hpp>
1616
1717#include < rmm/mr/per_device_resource.hpp>
1818#include < rmm/resource_ref.hpp>
Original file line number Diff line number Diff line change @@ -222,6 +222,22 @@ namespace *raft*
222222 :project: RAFT
223223 :members:
224224
225+ Dry-Run Resources
226+ ~~~~~~~~~~~~~~~~~
227+
228+ ``#include <raft/core/dry_run_resources.hpp> ``
229+
230+ namespace *raft *
231+
232+ .. doxygenclass :: raft::dry_run_resources
233+ :project: RAFT
234+ :members:
235+
236+ namespace *raft::util *
237+
238+ .. doxygenfunction :: raft::util::dry_run_execute
239+ :project: RAFT
240+
225241Device Properties
226242~~~~~~~~~~~~~~~~~
227243
Original file line number Diff line number Diff line change @@ -5,7 +5,7 @@ The dry run protocol lets callers estimate an algorithm's memory footprint witho
55## Using Dry Run Mode
66
77``` cpp
8- #include < raft/util /dry_run_resources.hpp>
8+ #include < raft/core /dry_run_resources.hpp>
99
1010raft::resources res;
1111// auto my_function(const raft::resources& res, my_args...);
You can’t perform that action at this time.
0 commit comments