================ @@ -0,0 +1,142 @@ +//===----------------------------------------------------------------------===// +// +// 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 +// +//===----------------------------------------------------------------------===// + +#ifndef _LIBCPP___ALGORITHM_PSTL_BACKENDS_OPENMP_BACKEND_TRANSFORM_H +#define _LIBCPP___ALGORITHM_PSTL_BACKENDS_OPENMP_BACKEND_TRANSFORM_H + +#include <__algorithm/pstl_backends/cpu_backends/backend.h> +#include <__algorithm/pstl_backends/openmp/backend.h> +#include <__algorithm/transform.h> +#include <__config> +#include <__iterator/concepts.h> +#include <__type_traits/is_execution_policy.h> +#include <optional> + +#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) +# pragma GCC system_header +#endif + +#if !defined(_LIBCPP_HAS_NO_INCOMPLETE_PSTL) && _LIBCPP_STD_VER >= 17 + +_LIBCPP_BEGIN_NAMESPACE_STD + +namespace __par_backend { +inline namespace __omp_gpu_backend { + +//===----------------------------------------------------------------------===// +// Templates for two iterators +//===----------------------------------------------------------------------===// + +template <class _Tp, class _DifferenceType, class _Up, class _Function> +_LIBCPP_HIDE_FROM_ABI optional<__empty> +__omp_parallel_for_simd(_Tp* __first1, _DifferenceType __n, _Up* __first2, _Function __f) noexcept { + __omp_gpu_backend::__omp_map_alloc(__first2, __n); + __omp_gpu_backend::__omp_map_to(__first1, __n); ---------------- ldionne wrote:
We should do `omp_map_to` before `omp_map_alloc` to make sure that things work if we are calling `transform` on a sequence and the output iterator is the sequence itself. We should add a test for that. https://github.com/llvm/llvm-project/pull/66968 _______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
