Skip to content

Commit 6405ec8

Browse files
Sergei-Lebedevkawashima-fj
authored andcommitted
fp16 hcoll bindings
Signed-off-by: Sergey Lebedev <[email protected]>
1 parent 1314516 commit 6405ec8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ompi/mca/coll/hcoll/coll_hcoll_dtypes.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,9 @@ static dte_data_representation_t* ompi_datatype_2_dte_data_rep[OMPI_DATATYPE_MAX
5757
&DTE_UINT32, /*OPAL_DATATYPE_UINT4 11 */
5858
&DTE_UINT64, /*OPAL_DATATYPE_UINT8 12 */
5959
&DTE_UINT128, /*OPAL_DATATYPE_UINT16 13 */
60-
&DTE_ZERO, /*OPAL_DATATYPE_FLOAT2 14 */
60+
#if defined(DTE_FLOAT16)
61+
&DTE_FLOAT16, /*OPAL_DATATYPE_FLOAT2 14 */
62+
#endif
6163
&DTE_FLOAT32, /*OPAL_DATATYPE_FLOAT4 15 */
6264
&DTE_FLOAT64, /*OPAL_DATATYPE_FLOAT8 16 */
6365
&DTE_FLOAT96, /*OPAL_DATATYPE_FLOAT12 17 */

0 commit comments

Comments
 (0)