Skip to content

Commit cedc56e

Browse files
committed
Merge pull request open-mpi#1099 from jladd-mlnx/topic/add-complex-dtype-support-hcoll-v2.1
Adds mapping to hcoll complex dte
2 parents d7c1bf4 + f730979 commit cedc56e

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

ompi/mca/coll/hcoll/coll_hcoll_dtypes.h

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@
2525
#define OPAL_DATATYPE_FLOAT8 16
2626
#define OPAL_DATATYPE_FLOAT12 17
2727
#define OPAL_DATATYPE_FLOAT16 18
28+
#define OPAL_DATATYPE_FLOAT_COMPLEX 19
29+
#define OPAL_DATATYPE_DOUBLE_COMPLEX 20
2830
2931
total 15 types
3032
*/
@@ -51,8 +53,13 @@ static dte_data_representation_t* ompi_datatype_2_dte_data_rep[OPAL_DATATYPE_MAX
5153
&DTE_FLOAT64, /*OPAL_DATATYPE_FLOAT8 16 */
5254
&DTE_FLOAT96, /*OPAL_DATATYPE_FLOAT12 17 */
5355
&DTE_FLOAT128, /*OPAL_DATATYPE_FLOAT16 18 */
56+
#if defined(DTE_FLOAT32_COMPLEX) && defined(DTE_FLOAT64_COMPLEX)
57+
&DTE_FLOAT32_COMPLEX, /*OPAL_DATATYPE_COMPLEX8 19 */
58+
&DTE_FLOAT64_COMPLEX, /*OPAL_DATATYPE_COMPLEX16 20 */
59+
#else
5460
&DTE_ZERO, /*OPAL_DATATYPE_COMPLEX8 19 */
5561
&DTE_ZERO, /*OPAL_DATATYPE_COMPLEX16 20 */
62+
#endif
5663
&DTE_ZERO, /*OPAL_DATATYPE_COMPLEX32 21 */
5764
&DTE_ZERO, /*OPAL_DATATYPE_BOOL 22 */
5865
&DTE_ZERO, /*OPAL_DATATYPE_WCHAR 23 */

0 commit comments

Comments
 (0)