Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Add BFloat16 support for radix sort #306

Merged
merged 2 commits into from
Jun 8, 2021
Merged

Conversation

zasdfgbnm
Copy link
Contributor

@zasdfgbnm zasdfgbnm commented May 12, 2021

No description provided.

@zasdfgbnm zasdfgbnm marked this pull request as ready for review May 12, 2021 22:37
@zasdfgbnm
Copy link
Contributor Author

Ready for review, tests pass.

@alliepiper
Copy link
Collaborator

This looks good to me. Thanks for taking the time to work on this!

Can squash the branch down to a single commit, and also change the comment about PyTorch in the test wrapper implementation? I know it's meant casually, but some internal legal folks may have an overreaction to that phrasing ;)

@zasdfgbnm
Copy link
Contributor Author

@allisonvacanti I have squashed this PR, and changed the comment to:

        // Refrence:
        // https://github.com/pytorch/pytorch/blob/44cc873fba5e5ffc4d4d4eef3bd370b653ce1ce1/c10/util/BFloat16.h#L51

@alliepiper
Copy link
Collaborator

Thanks -- I'll start tests soon.

@alliepiper
Copy link
Collaborator

Rebased and noted bfloat16 support in the docs. My doc change will conflict with #307, but I'll resolve that when merging this.

Starting tests now.

alliepiper added a commit to alliepiper/thrust that referenced this pull request May 20, 2021
@alliepiper
Copy link
Collaborator

gpuCI: NVIDIA/thrust#1439

DVS CL: 29986609

@alliepiper alliepiper added testing: internal ci in progress Currently testing on internal NVIDIA CI (DVS). testing: gpuCI in progress Started gpuCI testing. labels May 20, 2021
alliepiper added a commit to alliepiper/thrust that referenced this pull request May 21, 2021
@alliepiper alliepiper added testing: internal ci passed Passed internal NVIDIA CI (DVS). testing: gpuCI passed Passed gpuCI testing. and removed testing: internal ci in progress Currently testing on internal NVIDIA CI (DVS). testing: gpuCI in progress Started gpuCI testing. labels Jun 8, 2021
@alliepiper alliepiper merged commit 3cb3564 into NVIDIA:main Jun 8, 2021
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
testing: gpuCI passed Passed gpuCI testing. testing: internal ci passed Passed internal NVIDIA CI (DVS).
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants