File tree Expand file tree Collapse file tree 3 files changed +4
-4
lines changed
Expand file tree Collapse file tree 3 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -51,7 +51,7 @@ void atan2_kernel(TensorIterator& iter) {
5151 IPEX_DISPATCH_FLOATING_TYPES_AND2 (
5252 at::ScalarType::BFloat16,
5353 at::ScalarType::Half,
54- iter.dtype (),
54+ iter.common_dtype (),
5555 " atan2" ,
5656 [&]() {
5757 dpcpp_kernel_for_tensor_iter (
Original file line number Diff line number Diff line change @@ -17,15 +17,15 @@ namespace AtenIpexTypeXPU {
1717namespace impl {
1818
1919static void gcd_kernel_dpcpp (TensorIterator& iter) {
20- IPEX_DISPATCH_INTEGRAL_TYPES (iter.dtype (), " gcd" , [&]() {
20+ IPEX_DISPATCH_INTEGRAL_TYPES (iter.common_dtype (), " gcd" , [&]() {
2121 dpcpp_fast_mode_kernel_with_scalars (
2222 iter,
2323 [=](scalar_t a, scalar_t b) -> scalar_t { return calc_gcd (a, b); });
2424 });
2525}
2626
2727static void lcm_kernel_dpcpp (TensorIterator& iter) {
28- IPEX_DISPATCH_INTEGRAL_TYPES (iter.dtype (), " lcm" , [&]() {
28+ IPEX_DISPATCH_INTEGRAL_TYPES (iter.common_dtype (), " lcm" , [&]() {
2929 dpcpp_fast_mode_kernel_with_scalars (
3030 iter, [=](scalar_t a, scalar_t b) -> scalar_t {
3131 scalar_t g = calc_gcd (a, b);
Original file line number Diff line number Diff line change @@ -157,7 +157,7 @@ namespace AtenIpexTypeXPU {
157157 IPEX_DISPATCH_##types##_AND2 ( \
158158 at::ScalarType::Half, \
159159 at::ScalarType::BFloat16, \
160- iter.dtype (), \
160+ iter.common_dtype (), \
161161 #op, \
162162 [&]() { \
163163 dpcpp_kernel_with_scalars ( \
You can’t perform that action at this time.
0 commit comments