Skip to content

Commit ff44e11

Browse files
committed
Address RMMAllocator error allocating memory when running with realtime=false
Signed-off-by: Victor Chang <vicchang@nvidia.com>
1 parent 1302671 commit ff44e11

File tree

2 files changed

+11
-3
lines changed

2 files changed

+11
-3
lines changed

applications/distributed/ucx/ucx_endoscopy_tool_tracking/cpp/main.cpp

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,10 @@ class VideoInputFragment : public holoscan::Fragment {
4040

4141
auto replayer = make_operator<ops::VideoStreamReplayerOp>(
4242
"replayer",
43-
Arg("allocator", make_resource<RMMAllocator>("video_replayer_allocator")),
43+
Arg("allocator",
44+
make_resource<RMMAllocator>("video_replayer_allocator",
45+
Arg("device_memory_max_size") = std::string("256MB"),
46+
Arg("device_memory_initial_size") = std::string("256MB"))),
4447
from_config("replayer"),
4548
args);
4649

@@ -207,7 +210,12 @@ int main(int argc, char** argv) {
207210
auto app = holoscan::make_application<App>();
208211
app->config(config_path);
209212
app->set_datapath(data_directory);
210-
app->run();
211213

214+
auto trackers = app->track_distributed();
215+
app->run();
216+
for (const auto& [name, tracker] : trackers) {
217+
std::cout << "Fragment: " << name << std::endl;
218+
tracker->print();
219+
}
212220
return 0;
213221
}

applications/distributed/ucx/ucx_endoscopy_tool_tracking/cpp/metadata.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
]
3333
},
3434
"run": {
35-
"command": "<holohub_app_bin>/ucx_endoscopy_tool_tracking",
35+
"command": "<holohub_app_bin>/ucx_endoscopy_tool_tracking --data <holohub_data_dir>/endoscopy",
3636
"workdir": "holohub_bin"
3737
}
3838
}

0 commit comments

Comments
 (0)