Skip to content

Commit 0d5a57d

Browse files
committed
Merge pull request #1558 from vspetrov/hcoll_complex_dtype_support
Adds mapping to hcoll complex data type
2 parents 490b538 + 21f1c57 commit 0d5a57d

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)