rocPRIM data type support

The following table shows the supported input and output datatypes.

Supported Input/Output Types

Type	Support
int8	✅
int16	✅
int32	✅
int64	✅
float8	❌
float16	✅
bfloat16	✅
tensorfloat32	❌
float32	✅
float64	✅