frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[dynamo] Support builtin complex with constant args (#160799)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160799
Approved by: https://github.com/guilhermeleobas, https://github.com/mlazos
This commit is contained in:
Rob Timpe
2025-08-18 21:30:42 +00:00
committed by PyTorch MergeBot
parent 66166cf1e7
commit 35e4d97e04
16 changed files with 45 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
torch/_dynamo/variables/builtin.py
View File

@ -1478,6 +1478,21 @@ class BuiltinVariable(VariableTracker):
call_int = _call_int_float
call_float = _call_int_float
def call_complex(self, tx: "InstructionTranslator", *args, **kwargs):
if self.constant_args(*args, **kwargs):
try:
c = complex(
*(arg.as_python_constant() for arg in args),
**{k: kwargs[k].as_python_constant() for k in kwargs},
)
except (TypeError, ValueError) as exc:
raise_observed_exception(
type(exc),
tx,
args=list(map(ConstantVariable.create, exc.args)),
)
return ConstantVariable(c)
def call_bool(self, tx: "InstructionTranslator", arg):
# Emulate `PyBool_Type.tp_vectorcall` which boils down to `PyObject_IsTrue`.
# https://github.com/python/cpython/blob/3.12/Objects/object.c#L1674-L1697