Describe the enhancement requested
FloatingPointTypes() currently contains only float32() and float64(), so float16 is also absent from NumericTypes() and PrimitiveTypes().
As a result, every kernel registration loop and test suite that iterates over those lists skips float16, and support for it has to be added type by type instead. Several kernel families explicitly reject HalfFloatType with Status::NotImplemented today.
float16 should probably be a member of FloatingPointTypes(), consistent with the other floating point types.
Component(s)
C++
Describe the enhancement requested
FloatingPointTypes()currently contains onlyfloat32()andfloat64(), sofloat16is also absent fromNumericTypes()andPrimitiveTypes().As a result, every kernel registration loop and test suite that iterates over those lists skips
float16, and support for it has to be added type by type instead. Several kernel families explicitly rejectHalfFloatTypewithStatus::NotImplementedtoday.float16should probably be a member ofFloatingPointTypes(), consistent with the other floating point types.Component(s)
C++