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

Add size columns to an axes example.

Browse Source
This commit is contained in:
Allison Vacanti
2021-03-18 13:43:27 -04:00
parent 4e83e048ba
commit 52d3dfec8c
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/axes.cu
View File

@@ -142,10 +142,11 @@ void copy_type_conversion_sweep(nvbench::state &state,
// Number of InputTypes in 64 MiB:
const std::size_t num_values = 64 * 1024 * 1024 / sizeof(InputType);
// Report throughput stats:
state.add_element_count(num_values);
state.add_global_memory_reads<InputType>(num_values);
state.add_global_memory_writes<OutputType>(num_values);
// Report throughput stats: Passing an optional string adds a column to the
// output with the number of items/bytes.
state.add_element_count(num_values, "Items");
state.add_global_memory_reads<InputType>(num_values, "InSize");
state.add_global_memory_writes<OutputType>(num_values, "OutSize");
// Allocate device memory:
thrust::device_vector<InputType> in(num_values, 0);