File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
aten/src/ATen/native/cuda Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -111,7 +111,7 @@ void reciprocal_kernel_cuda(TensorIteratorBase& iter) {
111
111
};
112
112
113
113
auto either_inf = [](T real, T imag) {
114
- return ::isinf (real) || ::isinf (imag);
114
+ return ::isinf (real) || ::isinf (imag);
115
115
};
116
116
117
117
auto either_nan = [](T real, T imag) {
@@ -121,7 +121,7 @@ void reciprocal_kernel_cuda(TensorIteratorBase& iter) {
121
121
if (either_nan (v.real (), v.imag ()) || both_inf (v.real (), v.imag ())) {
122
122
// If either is Nan or both are infinite, return {nan, nan}
123
123
return {std::numeric_limits<T>::quiet_NaN (), std::numeric_limits<T>::quiet_NaN ()};
124
- } else if (either_inf (v.real (), v.imag ())) {
124
+ } else if (either_inf (v.real (), v.imag ())) {
125
125
// If either is Inf, return {0, 0}
126
126
return {0 , 0 };
127
127
}
@@ -143,7 +143,7 @@ void reciprocal_kernel_cuda(TensorIteratorBase& iter) {
143
143
});
144
144
});
145
145
#endif
146
- } else {
146
+ } else {
147
147
AT_DISPATCH_FLOATING_TYPES_AND2 (
148
148
ScalarType::Half, ScalarType::BFloat16,
149
149
dtype, " reciprocal_cuda" ,
You can’t perform that action at this time.
0 commit comments