NVIDIA / cccl

CUDA Core Compute Libraries
https://nvidia.github.io/cccl/
Other
1.3k stars 165 forks source link

[EPIC] Extended Floating-Point Support #31

Open jrhemstad opened 1 year ago

jrhemstad commented 1 year ago

We want to better support users in using extended floating-point types like __half and bfloat16 in our libraries.

One of the major projects would be adding new extended floating-point library types to libcu++ for things like __half and bfloat16. The main goals would be:

CUTLASS already has implementations of these types we could use as a starting point.

Tasks for new types

Tasks for existing __half/bfloat types

Tasks for extended floating-point vector types

cmpfeil commented 1 year ago

Would be eager to see this; in particular also specializations complex<__half> and complex<bfloat16>. Are there any plans on whether or when this might be available?