derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
160 lines
7.1 KiB
C
160 lines
7.1 KiB
C
/*===-- lib/runtime/complex-reduction.h -----------------------------*- C -*-===
|
|
*
|
|
* Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
* See https://llvm.org/LICENSE.txt for license information.
|
|
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
*
|
|
*===----------------------------------------------------------------------===*/
|
|
|
|
/* Wraps the C++-coded complex-valued SUM and PRODUCT reductions with
|
|
* C-coded wrapper functions returning _Complex values, to avoid problems
|
|
* with C++ build compilers that don't support C's _Complex.
|
|
*/
|
|
|
|
#ifndef FLANG_RT_RUNTIME_COMPLEX_REDUCTION_H_
|
|
#define FLANG_RT_RUNTIME_COMPLEX_REDUCTION_H_
|
|
|
|
#include "flang/Common/float128.h"
|
|
#include "flang/Runtime/entry-names.h"
|
|
#include <complex.h>
|
|
#include <stdbool.h>
|
|
|
|
struct CppDescriptor; /* dummy type name for Fortran::runtime::Descriptor */
|
|
|
|
#if defined(_MSC_VER) && !(defined(__clang_major__) && __clang_major__ >= 12)
|
|
typedef _Fcomplex float_Complex_t;
|
|
typedef _Dcomplex double_Complex_t;
|
|
typedef _Lcomplex long_double_Complex_t;
|
|
#else
|
|
typedef float _Complex float_Complex_t;
|
|
typedef double _Complex double_Complex_t;
|
|
typedef long double _Complex long_double_Complex_t;
|
|
#endif
|
|
|
|
#define REDUCTION_ARGS \
|
|
const struct CppDescriptor *x, const char *source, int line, int dim /*=0*/, \
|
|
const struct CppDescriptor *mask /*=NULL*/
|
|
#define REDUCTION_ARG_NAMES x, source, line, dim, mask
|
|
|
|
float_Complex_t RTNAME(SumComplex2)(REDUCTION_ARGS);
|
|
float_Complex_t RTNAME(SumComplex3)(REDUCTION_ARGS);
|
|
float_Complex_t RTNAME(SumComplex4)(REDUCTION_ARGS);
|
|
double_Complex_t RTNAME(SumComplex8)(REDUCTION_ARGS);
|
|
long_double_Complex_t RTNAME(SumComplex10)(REDUCTION_ARGS);
|
|
#if HAS_LDBL128 || HAS_FLOAT128
|
|
CFloat128ComplexType RTNAME(SumComplex16)(REDUCTION_ARGS);
|
|
#endif
|
|
|
|
float_Complex_t RTNAME(ProductComplex2)(REDUCTION_ARGS);
|
|
float_Complex_t RTNAME(ProductComplex3)(REDUCTION_ARGS);
|
|
float_Complex_t RTNAME(ProductComplex4)(REDUCTION_ARGS);
|
|
double_Complex_t RTNAME(ProductComplex8)(REDUCTION_ARGS);
|
|
long_double_Complex_t RTNAME(ProductComplex10)(REDUCTION_ARGS);
|
|
#if HAS_LDBL128 || HAS_FLOAT128
|
|
CFloat128ComplexType RTNAME(ProductComplex16)(REDUCTION_ARGS);
|
|
#endif
|
|
|
|
#define DOT_PRODUCT_ARGS \
|
|
const struct CppDescriptor *x, const struct CppDescriptor *y, \
|
|
const char *source, int line, int dim /*=0*/, \
|
|
const struct CppDescriptor *mask /*=NULL*/
|
|
#define DOT_PRODUCT_ARG_NAMES x, y, source, line, dim, mask
|
|
|
|
float_Complex_t RTNAME(DotProductComplex2)(DOT_PRODUCT_ARGS);
|
|
float_Complex_t RTNAME(DotProductComplex3)(DOT_PRODUCT_ARGS);
|
|
float_Complex_t RTNAME(DotProductComplex4)(DOT_PRODUCT_ARGS);
|
|
double_Complex_t RTNAME(DotProductComplex8)(DOT_PRODUCT_ARGS);
|
|
long_double_Complex_t RTNAME(DotProductComplex10)(DOT_PRODUCT_ARGS);
|
|
#if HAS_LDBL128 || HAS_FLOAT128
|
|
CFloat128ComplexType RTNAME(DotProductComplex16)(DOT_PRODUCT_ARGS);
|
|
#endif
|
|
|
|
#define REDUCE_ARGS(T, OP) \
|
|
OP operation, const struct CppDescriptor *x, const struct CppDescriptor *y, \
|
|
const char *source, int line, int dim /*=0*/, \
|
|
const struct CppDescriptor *mask /*=NULL*/, const T *identity /*=NULL*/, \
|
|
_Bool ordered /*=true*/
|
|
#define REDUCE_ARG_NAMES \
|
|
operation, x, y, source, line, dim, mask, identity, ordered
|
|
|
|
typedef float_Complex_t (*float_Complex_t_ref_op)(
|
|
const float_Complex_t *, const float_Complex_t *);
|
|
typedef float_Complex_t (*float_Complex_t_value_op)(
|
|
float_Complex_t, float_Complex_t);
|
|
typedef double_Complex_t (*double_Complex_t_ref_op)(
|
|
const double_Complex_t *, const double_Complex_t *);
|
|
typedef double_Complex_t (*double_Complex_t_value_op)(
|
|
double_Complex_t, double_Complex_t);
|
|
typedef long_double_Complex_t (*long_double_Complex_t_ref_op)(
|
|
const long_double_Complex_t *, const long_double_Complex_t *);
|
|
typedef long_double_Complex_t (*long_double_Complex_t_value_op)(
|
|
long_double_Complex_t, long_double_Complex_t);
|
|
|
|
float_Complex_t RTNAME(ReduceComplex2Ref)(
|
|
REDUCE_ARGS(float_Complex_t, float_Complex_t_ref_op));
|
|
float_Complex_t RTNAME(ReduceComplex2Value)(
|
|
REDUCE_ARGS(float_Complex_t, float_Complex_t_value_op));
|
|
float_Complex_t RTNAME(ReduceComplex3Ref)(
|
|
REDUCE_ARGS(float_Complex_t, float_Complex_t_ref_op));
|
|
float_Complex_t RTNAME(ReduceComplex3Value)(
|
|
REDUCE_ARGS(float_Complex_t, float_Complex_t_value_op));
|
|
float_Complex_t RTNAME(ReduceComplex4Ref)(
|
|
REDUCE_ARGS(float_Complex_t, float_Complex_t_ref_op));
|
|
float_Complex_t RTNAME(ReduceComplex4Value)(
|
|
REDUCE_ARGS(float_Complex_t, float_Complex_t_value_op));
|
|
double_Complex_t RTNAME(ReduceComplex8Ref)(
|
|
REDUCE_ARGS(double_Complex_t, double_Complex_t_ref_op));
|
|
double_Complex_t RTNAME(ReduceComplex8Value)(
|
|
REDUCE_ARGS(double_Complex_t, double_Complex_t_value_op));
|
|
long_double_Complex_t RTNAME(ReduceComplex10Ref)(
|
|
REDUCE_ARGS(long_double_Complex_t, long_double_Complex_t_ref_op));
|
|
long_double_Complex_t RTNAME(ReduceComplex10Value)(
|
|
REDUCE_ARGS(long_double_Complex_t, long_double_Complex_t_value_op));
|
|
#if HAS_LDBL128 || HAS_FLOAT128
|
|
typedef CFloat128ComplexType (*CFloat128ComplexType_ref_op)(
|
|
const CFloat128ComplexType *, const CFloat128ComplexType *);
|
|
typedef CFloat128ComplexType (*CFloat128ComplexType_value_op)(
|
|
CFloat128ComplexType, CFloat128ComplexType);
|
|
CFloat128ComplexType RTNAME(ReduceComplex16Ref)(
|
|
REDUCE_ARGS(CFloat128ComplexType, CFloat128ComplexType_ref_op));
|
|
CFloat128ComplexType RTNAME(ReduceComplex16Value)(
|
|
REDUCE_ARGS(CFloat128ComplexType, CFloat128ComplexType_value_op));
|
|
#endif
|
|
|
|
#define REDUCE_DIM_ARGS(T, OP) \
|
|
struct CppDescriptor *result, OP operation, const struct CppDescriptor *x, \
|
|
const struct CppDescriptor *y, const char *source, int line, int dim, \
|
|
const struct CppDescriptor *mask /*=NULL*/, const T *identity /*=NULL*/, \
|
|
_Bool ordered /*=true*/
|
|
#define REDUCE_DIM_ARG_NAMES \
|
|
result, operation, x, y, source, line, dim, mask, identity, ordered
|
|
|
|
void RTNAME(ReduceComplex2DimRef)(
|
|
REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_ref_op));
|
|
void RTNAME(ReduceComplex2DimValue)(
|
|
REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_value_op));
|
|
void RTNAME(ReduceComplex3DimRef)(
|
|
REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_ref_op));
|
|
void RTNAME(ReduceComplex3DimValue)(
|
|
REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_value_op));
|
|
void RTNAME(ReduceComplex4DimRef)(
|
|
REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_ref_op));
|
|
void RTNAME(ReduceComplex4DimValue)(
|
|
REDUCE_DIM_ARGS(float_Complex_t, float_Complex_t_value_op));
|
|
void RTNAME(ReduceComplex8DimRef)(
|
|
REDUCE_DIM_ARGS(double_Complex_t, double_Complex_t_ref_op));
|
|
void RTNAME(ReduceComplex8DimValue)(
|
|
REDUCE_DIM_ARGS(double_Complex_t, double_Complex_t_value_op));
|
|
void RTNAME(ReduceComplex10DimRef)(
|
|
REDUCE_DIM_ARGS(long_double_Complex_t, long_double_Complex_t_ref_op));
|
|
void RTNAME(ReduceComplex10DimValue)(
|
|
REDUCE_DIM_ARGS(long_double_Complex_t, long_double_Complex_t_value_op));
|
|
#if HAS_LDBL128 || HAS_FLOAT128
|
|
void RTNAME(ReduceComplex16DimRef)(
|
|
REDUCE_DIM_ARGS(CFloat128ComplexType, CFloat128ComplexType_ref_op));
|
|
void RTNAME(ReduceComplex16DimValue)(
|
|
REDUCE_DIM_ARGS(CFloat128ComplexType, CFloat128ComplexType_value_op));
|
|
#endif
|
|
|
|
#endif // FLANG_RT_RUNTIME_COMPLEX_REDUCTION_H_
|