Add keep option to distinct nvbench #16497
Conversation
bdice
added
improvement
| .add_int64_axis("NumRows", {10'000, 100'000, 1'000'000, 10'000'000}); | ||
| .add_string_axis("keep", {"any", "first", "last", "none"}) | ||
| .add_int64_axis("cardinality", | ||
| {100, 1'000, 10'000, 100'000, 1'000'000, 10'000'000, 100'000'000, 1'000'000'000}) |
There was a problem hiding this comment.
We can just decrease the default values in the axis.
| if (cardinality > num_rows) { | ||
| state.skip("cardinality > num_rows"); | ||
| return; | ||
| } |
There was a problem hiding this comment.
I would prefer to omit this skipping condition. I recognize that we can't have 1M distinct elements in 1K rows, but this condition adds a lot of friction when sweeping NumRows for the high cardinality case. It forces me to run a full factorial of matching NumRows and Cardinality values and filter the outputs for the highest Cardinality unskipped for each NumRows.
There was a problem hiding this comment.
I'll rewrite this logic. Thanks for the feedback!
There was a problem hiding this comment.
Hmm. @GregoryKimball I reviewed the NVBench docs and I don't see a way to filter out certain jobs except by skipping them. https://github.com/NVIDIA/nvbench/blob/main/docs/benchmarks.md#beware-combinatorial-explosion-is-lurking
We might be able to use a string axis like {"100,100", "100,1000", ..., "1000000000,1000000000"} and parse it, but that's hard to maintain.
There was a problem hiding this comment.
I don't see a way to filter out certain jobs except by skipping them.
NVIDIA/nvbench#80 can solve this issue but the PR has been stalled for a while.
Co-authored-by: Yunsong Wang <yunsongw@nvidia.com>
|
Thanks guys, here is a performance snapshot of these benchmarks on A100 I would like to merge this update ASAP and then have @srinivasyadav18 pull the changes into #16484 I'm noticing that these throughput numbers on A100 are about 10x lower than what @srinivasyadav18 posted for H100. This is another reason I'm interested in running wider benchmarks on #16484 |
…inct-benchmark-keep
|
Just fixed the CI style check. I'll merge this when CI passes. |
|
/merge |
Description
This PR adopts some work from @srinivasyadav18 with additional modifications. This is meant to complement #16484.
Checklist