Skip to content

Adds tests for large number of items for cub::DeviceSelect#1612

Merged
elstehle merged 7 commits intoNVIDIA:mainfrom
elstehle:enh/test-large-select
Apr 14, 2024
Merged

Adds tests for large number of items for cub::DeviceSelect#1612
elstehle merged 7 commits intoNVIDIA:mainfrom
elstehle:enh/test-large-select

Conversation

@elstehle
Copy link
Contributor

Description

In PR #1379, we consolidated thrust::copy_if to use cub::DeviceSelect. Along with that change, we made thrust to dispatch using offset types using i64 if the given number of items was exceeding INT_MAX.

Testing for large number of items in thrust::copy_if has been limited to some sanity tests.

We want to have more elaborate test coverage by adding tests for large number of items to the CUB tests of cub::DeviceSelect, since we have better testing infrastructure in CUB.

This PR adds those tests.

Closes #1584

Checklist

  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.

@elstehle elstehle requested review from a team as code owners April 10, 2024 16:09
Copy link
Contributor

@alliepiper alliepiper left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A couple minor suggestions inline.

Merging main into this should fix the CI issues as well.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Archived in project

Development

Successfully merging this pull request may close these issues.

Add tests for large number of items to DeviceSelect

3 participants