Hugging Face's logo

camenduru
/
thrust

Model card Files
xet
 Community
thrust / build /_deps /metal-src /example /src /lambda.cpp
camenduru's picture
camenduru
thanks to nvidia ❤
8ae5fc5
Raw History Blame Contribute Delete
3.92 kB
#include <metal.hpp>
#include "example.hpp"
#include <memory>
#include <tuple>
#include <type_traits>
#if !defined(METAL_WORKAROUND)
HIDE(
/// [lbd1]
template<class T>
using expr = T*;
using lbd = metal::lambda<expr>;
/// [lbd1]
IS_SAME(metal::is_lambda<lbd>, metal::true_);
)
HIDE(
/// [is_lambda]
template<class T>
using expr = T*;
IS_SAME(metal::is_lambda<void>, metal::false_);
IS_SAME(metal::is_lambda<metal::lambda<expr>>, metal::true_);
IS_SAME(metal::is_lambda<metal::trait<std::is_pointer>>, metal::true_);
IS_SAME(metal::is_lambda<metal::lazy<std::add_pointer>>, metal::true_);
/// [is_lambda]
)
#endif
HIDE(
/// [as_lambda]
IS_SAME(metal::as_lambda<std::shared_ptr<int>>, metal::lambda<std::shared_ptr>);
IS_SAME(metal::as_lambda<std::unique_ptr<int>>, metal::lambda<std::unique_ptr>);
IS_SAME(metal::as_lambda<std::tuple<int, char, float>>, metal::lambda<std::tuple>);
/// [as_lambda]
)
HIDE(
/// [trait]
IS_SAME(metal::invoke<metal::trait<std::is_fundamental>, int>, metal::true_);
IS_SAME(metal::invoke<metal::trait<std::is_fundamental>, void>, metal::true_);
IS_SAME(metal::invoke<metal::trait<std::is_integral>, int>, metal::true_);
IS_SAME(metal::invoke<metal::trait<std::is_integral>, void>, metal::false_);
IS_SAME(metal::invoke<metal::trait<std::is_class>, int>, metal::false_);
IS_SAME(metal::invoke<metal::trait<std::is_class>, void>, metal::false_);
IS_SAME(metal::invoke<metal::trait<std::is_convertible>, int*, void*>, metal::true_);
IS_SAME(metal::invoke<metal::trait<std::is_convertible>, void*, int*>, metal::false_);
/// [trait]
)
HIDE(
/// [lazy]
IS_SAME(metal::invoke<metal::lazy<std::add_pointer>, int>, int*);
IS_SAME(metal::invoke<metal::lazy<std::decay>, int()>, int(*)());
IS_SAME(metal::invoke<metal::lazy<std::common_type>, int[], void*>, void*);
/// [lazy]
)
HIDE(
/// [is_invocable]
template<class val>
using array = metal::identity<val[]>; // MSVC friendly
using lbd = metal::lambda<array>;
IS_SAME(metal::is_invocable<lbd, int, int>, metal::false_); // arity mismatch
IS_SAME(metal::is_invocable<lbd, void>, metal::false_); // void[] is ill-formed
IS_SAME(metal::is_invocable<lbd, int>, metal::true_);
/// [is_invocable]
)
HIDE(
/// [always]
using void_ = metal::always<void>;
IS_SAME(metal::invoke<void_>, void);
IS_SAME(metal::invoke<void_, bool, char, long, float>, void);
IS_SAME(metal::invoke<metal::always<void_>, bool, char, long, float>, void_);
/// [always]
)
HIDE(
/// [arg]
IS_SAME(metal::invoke<metal::arg<1>, bool, char, long, float>, bool);
IS_SAME(metal::invoke<metal::arg<2>, bool, char, long, float>, char);
IS_SAME(metal::invoke<metal::arg<3>, bool, char, long, float>, long);
IS_SAME(metal::invoke<metal::arg<4>, bool, char, long, float>, float);
/// [arg]
)
#if !defined(METAL_WORKAROUND)
HIDE(
/// [invoke]
using lbd = metal::lazy<std::common_type>;
IS_SAME(metal::invoke<lbd, bool, char, long, float>, float);
/// [invoke]
)
HIDE(
/// [apply]
using lbd = metal::lazy<std::common_type>;
IS_SAME(metal::apply<lbd, metal::list<bool, char, long, float>>, float);
/// [apply]
)
HIDE(
/// [partial]
using promote = metal::partial<metal::lazy<std::common_type>, int>;
IS_SAME(metal::invoke<promote, char>, int);
IS_SAME(metal::invoke<promote, short>, int);
IS_SAME(metal::invoke<promote, long>, long);
/// [partial]
)
HIDE(
/// [bind]
using promote = metal::bind<
metal::lazy<std::common_type>,
metal::always<int>,
metal::_1 // equivalent to metal::arg<1>
>;
IS_SAME(metal::invoke<promote, char>, int);
IS_SAME(metal::invoke<promote, short>, int);
IS_SAME(metal::invoke<promote, long>, long);
using uac = metal::bind< // usual arithmetic conversion
metal::lazy<std::common_type>,
metal::bind<promote, metal::_1>,
metal::bind<promote, metal::_2>
>;
IS_SAME(metal::invoke<uac, int, long>, long);
IS_SAME(metal::invoke<uac, char, char>, int);
IS_SAME(metal::invoke<uac, short, float>, float);
/// [bind]
)
#endif