File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
dpctl/tensor/libtensor/include/kernels Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -50,12 +50,18 @@ namespace tensor
50
50
namespace kernels
51
51
{
52
52
53
+ template <typename ReductionOpT, typename T> struct needs_workaround
54
+ {
55
+ static constexpr bool value =
56
+ std::is_same_v<ReductionOpT, sycl::multiplies<T>> &&
57
+ (std::is_same_v<T, std::int64_t > || std::is_same_v<T, std::uint64_t >);
58
+ };
59
+
53
60
template <typename ReductionOpT, typename T> struct can_use_reduce_over_group
54
61
{
55
62
static constexpr bool value =
56
63
sycl::has_known_identity<ReductionOpT, T>::value &&
57
- !std::is_same_v<T, std::int64_t > && !std::is_same_v<T, std::uint64_t > &&
58
- !std::is_same_v<ReductionOpT, sycl::multiplies<T>>;
64
+ !needs_workaround<ReductionOpT, T>::value;
59
65
};
60
66
61
67
template <typename argT,
You can’t perform that action at this time.
0 commit comments