/* * Copyright 2008-2018 NVIDIA Corporation * * 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. */ #pragma once #include #include #if THRUST_CPP_DIALECT >= 2011 #include #include #include THRUST_NAMESPACE_BEGIN namespace detail { // We need a way to compute the return type of `select_system`, which is found // by using `thrust::system::detail::generic::select_system` and then making an // ADL call. We have no trait that defines the return type. With the // limitations of C++11 return type deduction, we need to be able to stick all // of that into `decltype`. So, we put the using statement into a detail // namespace, and then implement the generic dispatch function in that // namespace. namespace select_system_detail { using thrust::system::detail::generic::select_system; struct select_system_fn final { __thrust_exec_check_disable__ template __host__ __device__ auto operator()( thrust::detail::execution_policy_base const& exec0 ) const THRUST_DECLTYPE_RETURNS( select_system( thrust::detail::derived_cast(thrust::detail::strip_const(exec0)) ) ) __thrust_exec_check_disable__ template __host__ __device__ auto operator()( thrust::detail::execution_policy_base const& exec0 , thrust::detail::execution_policy_base const& exec1 ) const THRUST_DECLTYPE_RETURNS( select_system( thrust::detail::derived_cast(thrust::detail::strip_const(exec0)) , thrust::detail::derived_cast(thrust::detail::strip_const(exec1)) ) ) }; } // namespace select_system_detail THRUST_INLINE_CONSTANT select_system_detail::select_system_fn select_system{}; } // detail THRUST_NAMESPACE_END #endif // THRUST_CPP_DIALECT >= 2011