mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Add ScalarList overload to _foreach_lerp
(#134482)
Related: - https://github.com/pytorch/pytorch/issues/133367 Pull Request resolved: https://github.com/pytorch/pytorch/pull/134482 Approved by: https://github.com/janeyx99
This commit is contained in:
committed by
PyTorch MergeBot
parent
7624d625c0
commit
6a368b3fc5
@ -1534,6 +1534,8 @@ def check_autodiff_sample(op, sample, dtype, is_inplace):
|
||||
or (isinstance(sample.args[-1], complex))
|
||||
)
|
||||
if rhs_arg_has_complex_number and dtype == torch.float64:
|
||||
if op.name == "_foreach_lerp":
|
||||
return False, "value cannot be converted to type double without overflow"
|
||||
if op.name in (
|
||||
"_foreach_clamp_max",
|
||||
"_foreach_clamp_min",
|
||||
|
Reference in New Issue
Block a user