File size: 1,541 Bytes
8ae5fc5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
#pragma once
#include <iostream>
#include <string>
#include <typeinfo>
#include <unittest/system.h>
#include <thrust/extrema.h>
#include <thrust/limits.h>
#include <thrust/detail/type_traits.h>
namespace unittest
{
template<typename T>
std::string type_name(void)
{
return demangle(typeid(T).name());
} // end type_name()
// Use this with counting_iterator to avoid generating a range larger than we
// can represent.
template <typename T>
typename THRUST_NS_QUALIFIER::detail::disable_if<
THRUST_NS_QUALIFIER::detail::is_floating_point<T>::value
, T
>::type truncate_to_max_representable(std::size_t n)
{
return static_cast<T>(THRUST_NS_QUALIFIER::min<std::size_t>(
n,
static_cast<std::size_t>(THRUST_NS_QUALIFIER::numeric_limits<T>::max())));
}
// TODO: This probably won't work for `half`.
template <typename T>
typename THRUST_NS_QUALIFIER::detail::enable_if<
THRUST_NS_QUALIFIER::detail::is_floating_point<T>::value
, T
>::type truncate_to_max_representable(std::size_t n)
{
return THRUST_NS_QUALIFIER::min<T>(
static_cast<T>(n),
THRUST_NS_QUALIFIER::numeric_limits<T>::max());
}
} // end unittest
template <typename Iterator>
void PRINT(Iterator first, Iterator last)
{
size_t n = 0;
for (Iterator i = first; i != last; i++, n++)
std::cout << ">>> [" << n << "] = " << *i << std::endl;
}
template <typename Container>
void PRINT(const Container& c)
{
PRINT(c.begin(), c.end());
}
template <size_t N>
void PRINT(const char (&c)[N])
{
std::cout << std::string(c, c + N) << std::endl;
}
|