-
Notifications
You must be signed in to change notification settings - Fork 915
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Convert compression and io to string axis type in IO benchmarks #14347
Convert compression and io to string axis type in IO benchmarks #14347
Conversation
Signed-off-by: Suraj Aralihalli <[email protected]>
Signed-off-by: Suraj Aralihalli <[email protected]>
/ok to test |
Signed-off-by: Suraj Aralihalli <[email protected]>
Signed-off-by: Suraj Aralihalli <[email protected]>
This modification allows for the control of the io and compression types in
|
@GregoryKimball Can you review the PR please, I can update other benchmark types, if this looks right to you! |
Thank you @SurajAralihalli for this change. I recommend just starting with the parquet reader benchmarks to begin with. @vuule and I were discussing adding a V1/V2 header parameter as well, but that can be in a follow-on PR. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good, thank you!
Signed-off-by: Suraj Aralihalli <[email protected]>
Sorry for my delayed response. @vuule If this pull request aligns with the existing requirements please feel free to run the tests and merge the pull request. |
/ok to test |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This should reduce the compile time slightly too since the template arguments are reduced.
/ok to test |
Signed-off-by: Suraj Aralihalli <[email protected]>
…f into convert_io_axis_type
Signed-off-by: Suraj Aralihalli <[email protected]>
/ok to test |
Signed-off-by: Suraj Aralihalli <[email protected]>
Signed-off-by: Suraj Aralihalli <[email protected]>
/ok to test |
/merge |
…dsai#14347) Addresses issue: [rapidsai#12739](rapidsai#12739) This PR transforms compression and io into string axis types to enable the selection of different values via the CLI, eliminating the need to execute all values in an automation when required. Additionally, this PR introduces two new functions, `retrieve_io_type_enum` and `retrieve_compression_type_enum`, which facilitate the conversion of string input into the corresponding enum type that can be used in benchmarking functions. IO Benchmarks: - [x] PARQUET READER For example: `./PARQUET_READER_NVBENCH -b parquet_read_io_compression --axis io_type=[HOST_BUFFER] --axis compression_type=[NONE]` Authors: - Suraj Aralihalli (https://github.com/SurajAralihalli) Approvers: - Vukasin Milovanovic (https://github.com/vuule) - Nghia Truong (https://github.com/ttnghia) URL: rapidsai#14347
Addresses issue: #12739
This PR transforms compression and io into string axis types to enable the selection of different values via the CLI, eliminating the need to execute all values in an automation when required. Additionally, this PR introduces two new functions,
retrieve_io_type_enum
andretrieve_compression_type_enum
, which facilitate the conversion of string input into the corresponding enum type that can be used in benchmarking functions.IO Benchmarks:
For example:
./PARQUET_READER_NVBENCH -b parquet_read_io_compression --axis io_type=[HOST_BUFFER] --axis compression_type=[NONE]
Checklist