NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 14:58:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Only print actively used devices when running benchmarks.

Browse Source
This commit is contained in:
Allison Vacanti
2021-10-07 17:11:50 -04:00
parent fcb9277b1d
commit 3c5f445ef0
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
nvbench/markdown_printer.cu
View File

@@ -45,7 +45,10 @@ void markdown_printer::do_print_device_info()
fmt::memory_buffer buffer;
fmt::format_to(buffer, "# Devices\n\n");
const auto &devices = nvbench::device_manager::get().get_devices();
const auto &device_mgr = nvbench::device_manager::get();
const auto &devices = device_mgr.get_number_of_used_devices() > 0
? device_mgr.get_used_devices()
: device_mgr.get_devices();
for (const auto &device : devices)
{
const auto [gmem_free, gmem_used] = device.get_global_memory_usage();