Enabled more data types for oneMKL gemm_batch (#8236)
This commit is contained in:
parent
148ec970b6
commit
1f3e1b66e2
2 changed files with 2 additions and 32 deletions
|
@ -2426,6 +2426,7 @@ namespace dpct
|
|||
b, ldb, beta, c, ldc, batch_size);
|
||||
break;
|
||||
}
|
||||
#endif
|
||||
case detail::get_type_combination_id(
|
||||
library_data_t::real_int8, library_data_t::real_int8,
|
||||
library_data_t::real_int32, library_data_t::real_int32):
|
||||
|
@ -2458,7 +2459,6 @@ namespace dpct
|
|||
batch_size);
|
||||
break;
|
||||
}
|
||||
#endif
|
||||
case detail::get_type_combination_id(
|
||||
library_data_t::real_half, library_data_t::real_half,
|
||||
library_data_t::real_half, library_data_t::real_float):
|
||||
|
@ -2595,6 +2595,7 @@ namespace dpct
|
|||
stride_c, batch_size);
|
||||
break;
|
||||
}
|
||||
#endif
|
||||
case detail::get_type_combination_id(
|
||||
library_data_t::real_int8, library_data_t::real_int8,
|
||||
library_data_t::real_int32, library_data_t::real_int32):
|
||||
|
@ -2623,7 +2624,6 @@ namespace dpct
|
|||
beta, c, ldc, stride_c, batch_size);
|
||||
break;
|
||||
}
|
||||
#endif
|
||||
case detail::get_type_combination_id(
|
||||
library_data_t::real_half, library_data_t::real_half,
|
||||
library_data_t::real_half, library_data_t::real_float):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue