We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent abf8b6b commit 79cad09Copy full SHA for 79cad09
advanced_source/dispatcher.rst
@@ -293,7 +293,8 @@ the safest choice for the execution type:
293
294
Tensor my_multiple_input_op_autocast(const Tensor& t0, const Tensor& t1) {
295
c10::impl::ExcludeDispatchKeyGuard no_autocast(c10::DispatchKey::Autocast);
296
- auto exec_type = at::autocast::promote_type(at::kHalf/*optimistic initial guess*/, t0, t1);
+ // The required at::kHalf argument is an optimistic initial guess.
297
+ auto exec_type = at::autocast::promote_type(at::kHalf, t0, t1);
298
return my_multiple_input_op(at::autocast::cached_cast(exec_type, t0),
299
at::autocast::cached_cast(exec_type, t1));
300
}
0 commit comments