execution.h

Go to the documentation of this file.

/*
 *  Copyright 2022 Patrick Stotko
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
 
#ifndef STDGPU_EXECUTION_H
#define STDGPU_EXECUTION_H
 
#include <type_traits>
 
#include <stdgpu/platform.h>
#include <stdgpu/type_traits.h>
 
#include <thrust/execution_policy.h>
 
namespace stdgpu
{
 
template <typename T>
struct is_execution_policy : std::is_base_of<thrust::execution_policy<T>, T>
{
};
 
template <typename T>
inline constexpr bool is_execution_policy_v = is_execution_policy<T>::value;
 
} // namespace stdgpu
 
namespace stdgpu::execution
{
 
template <typename T>
using execution_policy = thrust::execution_policy<T>;
 
using device_policy = std::remove_const_t<decltype(thrust::device)>;
 
static_assert(is_execution_policy_v<remove_cvref_t<device_policy>>,
              "stdgpu::execution::device_policy: Should be an execution policy but is not");
 
using host_policy = std::remove_const_t<decltype(thrust::host)>;
 
static_assert(is_execution_policy_v<remove_cvref_t<host_policy>>,
              "stdgpu::execution::host_policy: Should be an execution policy but is not");
 
constexpr device_policy device;
 
constexpr host_policy host;
 
} // namespace stdgpu::execution
 
// execution_detail.cuh declares device-qualified functions, so include it only
// when the device compiler is active (host-only translation units, e.g. plain
// .cpp files in the CUDA backend, must not see the __device__ declarations).
#if STDGPU_DETAIL_IS_DEVICE_COMPILED
    #include <stdgpu/impl/execution_detail.cuh>
#endif
 
#endif // STDGPU_EXECUTION_H