GOSTCoin CUDA miner project, compatible with most nvidia cards, containing only gostd algo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

278 lines
7.8 KiB

/*
* Copyright 2008-2012 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 <thrust/detail/config.h>
#include <thrust/detail/raw_reference_cast.h>
#include <thrust/detail/type_traits.h>
#include <thrust/detail/tuple_transform.h>
namespace thrust
{
namespace detail
{
// specialize is_unwrappable
// a tuple is_unwrappable if any of its elements is_unwrappable
template<
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
struct is_unwrappable<
thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>
: or_<
is_unwrappable<T0>,
is_unwrappable<T1>,
is_unwrappable<T2>,
is_unwrappable<T3>,
is_unwrappable<T4>,
is_unwrappable<T5>,
is_unwrappable<T6>,
is_unwrappable<T7>,
is_unwrappable<T8>,
is_unwrappable<T9>
>
{};
// specialize is_unwrappable
// a tuple_of_iterator_references is_unwrappable if any of its elements is_unwrappable
template<
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
struct is_unwrappable<
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>
: or_<
is_unwrappable<T0>,
is_unwrappable<T1>,
is_unwrappable<T2>,
is_unwrappable<T3>,
is_unwrappable<T4>,
is_unwrappable<T5>,
is_unwrappable<T6>,
is_unwrappable<T7>,
is_unwrappable<T8>,
is_unwrappable<T9>
>
{};
namespace raw_reference_detail
{
// unlike raw_reference,
// raw_reference_tuple_helper needs to return a value
// when it encounters one, rather than a reference
// upon encountering tuple, recurse
//
// we want the following behavior:
// 1. T -> T
// 2. T& -> T&
// 3. null_type -> null_type
// 4. reference<T> -> T&
// 5. tuple_of_iterator_references<T> -> tuple_of_iterator_references<raw_reference_tuple_helper<T>::type>
// wrapped references are unwrapped using raw_reference, otherwise, return T
template<typename T>
struct raw_reference_tuple_helper
: eval_if<
is_unwrappable<
typename remove_cv<T>::type
>::value,
raw_reference<T>,
identity_<T>
>
{};
// recurse on tuples
template <
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
struct raw_reference_tuple_helper<
thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>
{
typedef thrust::tuple<
typename raw_reference_tuple_helper<T0>::type,
typename raw_reference_tuple_helper<T1>::type,
typename raw_reference_tuple_helper<T2>::type,
typename raw_reference_tuple_helper<T3>::type,
typename raw_reference_tuple_helper<T4>::type,
typename raw_reference_tuple_helper<T5>::type,
typename raw_reference_tuple_helper<T6>::type,
typename raw_reference_tuple_helper<T7>::type,
typename raw_reference_tuple_helper<T8>::type,
typename raw_reference_tuple_helper<T9>::type
> type;
};
template <
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
struct raw_reference_tuple_helper<
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>
{
typedef thrust::detail::tuple_of_iterator_references<
typename raw_reference_tuple_helper<T0>::type,
typename raw_reference_tuple_helper<T1>::type,
typename raw_reference_tuple_helper<T2>::type,
typename raw_reference_tuple_helper<T3>::type,
typename raw_reference_tuple_helper<T4>::type,
typename raw_reference_tuple_helper<T5>::type,
typename raw_reference_tuple_helper<T6>::type,
typename raw_reference_tuple_helper<T7>::type,
typename raw_reference_tuple_helper<T8>::type,
typename raw_reference_tuple_helper<T9>::type
> type;
};
} // end raw_reference_detail
// if a tuple "tuple_type" is_unwrappable,
// then the raw_reference of tuple_type is a tuple of its members' raw_references
// else the raw_reference of tuple_type is tuple_type &
template <
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
struct raw_reference<
thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>
{
private:
typedef thrust::tuple<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> tuple_type;
public:
typedef typename eval_if<
is_unwrappable<tuple_type>::value,
raw_reference_detail::raw_reference_tuple_helper<tuple_type>,
add_reference<tuple_type>
>::type type;
};
template <
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
struct raw_reference<
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>
{
private:
typedef detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> tuple_type;
public:
typedef typename raw_reference_detail::raw_reference_tuple_helper<tuple_type>::type type;
// XXX figure out why is_unwrappable seems to be broken for tuple_of_iterator_references
//typedef typename eval_if<
// is_unwrappable<tuple_type>::value,
// raw_reference_detail::raw_reference_tuple_helper<tuple_type>,
// add_reference<tuple_type>
//>::type type;
};
struct raw_reference_caster
{
template<typename T>
__host__ __device__
typename detail::raw_reference<T>::type operator()(T &ref)
{
return thrust::raw_reference_cast(ref);
}
template<typename T>
__host__ __device__
typename detail::raw_reference<const T>::type operator()(const T &ref)
{
return thrust::raw_reference_cast(ref);
}
template<
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
__host__ __device__
typename detail::raw_reference<
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>::type
operator()(thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> t,
typename enable_if<
is_unwrappable<thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> >::value
>::type * = 0)
{
return thrust::raw_reference_cast(t);
}
}; // end raw_reference_caster
} // end detail
template<
typename T0, typename T1, typename T2,
typename T3, typename T4, typename T5,
typename T6, typename T7, typename T8,
typename T9
>
__host__ __device__
typename detail::enable_if_unwrappable<
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>,
typename detail::raw_reference<
thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9>
>::type
>::type
raw_reference_cast(thrust::detail::tuple_of_iterator_references<T0,T1,T2,T3,T4,T5,T6,T7,T8,T9> t)
{
thrust::detail::raw_reference_caster f;
// note that we pass raw_reference_tuple_helper, not raw_reference as the unary metafunction
// the subtle difference is important
return thrust::detail::tuple_host_device_transform<detail::raw_reference_detail::raw_reference_tuple_helper>(t, f);
} // end raw_reference_cast
} // end thrust