Arg max functor - Because NVIDIA's hipcub::ArgMax doesn't work with bfloat16 (HOST-SIDE)
More...
#include <test_utils_argminmax.hpp>
|
|
template<class Key , class Value > |
| HIPCUB_HOST_DEVICE constexpr KeyValuePair< Key, Value > | operator() (const KeyValuePair< Key, Value > &a, const KeyValuePair< Key, Value > &b) const |
| |
|
template<typename OffsetT , class T , std::enable_if_t< std::is_same< T, test_utils::half >::value||std::is_same< T, test_utils::bfloat16 >::value, bool > = true> |
| HIPCUB_HOST_DEVICE __forceinline__ hipcub::KeyValuePair< OffsetT, T > | operator() (const hipcub::KeyValuePair< OffsetT, T > &a, const hipcub::KeyValuePair< OffsetT, T > &b) const |
| |
Arg max functor - Because NVIDIA's hipcub::ArgMax doesn't work with bfloat16 (HOST-SIDE)
The documentation for this struct was generated from the following files:
