saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update base for Update on "[dynamo][user-defined] Unify standard and non-standard __new__ codebase"

Browse source 
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx chenyang78 kadeng chauhang amjames

[ghstack-poisoned]
This commit is contained in:
Animesh Jain 2025-02-09 00:32:51 -08:00
parent 2fba86c772
commit 16ef1f542f
2 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
torch/_dynamo/variables/misc.py
View file

@ -164,10 +164,13 @@ class SuperVariable(VariableTracker):
and variables.UserDefinedClassVariable.is_supported_new_method(inner_fn)
):
user_cls = inner_fn.__self__
if hasattr(user_cls, "__module__") and user_cls.__module__ == "bulitins":
if hasattr(user_cls, "__module__") and user_cls.__module__ == "builtins":
user_cls_vt = variables.BuiltinVariable(user_cls)
else:
user_cls_vt = variables.UserDefinedClassVariable(user_cls)
user_cls_source = source.member
user_cls_vt = variables.UserDefinedClassVariable(
user_cls, source=user_cls_source
)
return user_cls_vt.call_method(tx, "__new__", args, kwargs)
elif isinstance(inner_fn, staticmethod) and isinstance(
inner_fn.__func__, types.FunctionType

11
torch/_dynamo/variables/user_defined.py
View file

@ -344,6 +344,17 @@ class UserDefinedClassVariable(UserDefinedVariable):
return variables.ConstantVariable(self.value == args[0].value)
elif name == "__ne__" and len(args) == 1 and hasattr(args[0], "value"):
return variables.ConstantVariable(self.value != args[0].value)
elif (
name == "__new__"
and self.value is collections.OrderedDict
and isinstance(args[0], UserDefinedClassVariable)
and args[0].value is collections.OrderedDict
):
assert len(args) == 1
assert len(kwargs) == 0
return variables.ConstDictVariable(
{}, collections.OrderedDict, mutation_type=ValueMutationNew()
)
elif name == "__new__" and UserDefinedClassVariable.is_supported_new_method(
self.value.__new__
):