Skip to content

Commit 1827858

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

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ompi/mca/coll/hcoll/coll_hcoll_dtypes.h

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,11 @@ 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+
#else
63+
&DTE_ZERO,
64+
#endif
6165
&DTE_FLOAT32, /*OPAL_DATATYPE_FLOAT4 15 */
6266
&DTE_FLOAT64, /*OPAL_DATATYPE_FLOAT8 16 */
6367
&DTE_FLOAT96, /*OPAL_DATATYPE_FLOAT12 17 */

0 commit comments

Comments
 (0)