Skip to content

Commit 39dd0da

Browse files
More specializations for hyperparameters of add_contig_impl
1 parent e3c341e commit 39dd0da

File tree

1 file changed

+20
-2
lines changed
  • dpctl/tensor/libtensor/include/kernels/elementwise_functions

1 file changed

+20
-2
lines changed

dpctl/tensor/libtensor/include/kernels/elementwise_functions/add.hpp

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -209,17 +209,35 @@ using vsu_ns::ContigHyperparameterSetDefault;
209209
template <typename argTy1, typename argTy2> struct AddContigHyperparameterSet
210210
{
211211
using value_type = typename std::disjunction<
212+
BinaryContigHyperparameterSetEntry<argTy1,
213+
std::int32_t,
214+
argTy2,
215+
std::int32_t,
216+
2u,
217+
2u>,
218+
BinaryContigHyperparameterSetEntry<argTy1,
219+
std::int32_t,
220+
argTy2,
221+
std::int32_t,
222+
2u,
223+
2u>,
212224
BinaryContigHyperparameterSetEntry<argTy1,
213225
std::int64_t,
214226
argTy2,
215227
std::int64_t,
216-
1u,
228+
2u,
217229
2u>,
218230
BinaryContigHyperparameterSetEntry<argTy1,
219231
std::uint64_t,
220232
argTy2,
221233
std::uint64_t,
222-
1u,
234+
2u,
235+
2u>,
236+
BinaryContigHyperparameterSetEntry<argTy1,
237+
float,
238+
argTy2,
239+
float,
240+
2u,
223241
2u>,
224242
BinaryContigHyperparameterSetEntry<argTy1,
225243
double,

0 commit comments

Comments
 (0)