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 on "[dynamo][lists] Support list subclasses"

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 23:45:21 -08:00
parent 7270a4a6de
commit 8fbc9df45e
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
torch/_dynamo/source.py
View file

@ -559,6 +559,7 @@ class DictGetItemSource(ChainedSource):
self.index.reconstruct(codegen)
else:
codegen.append_output(codegen.create_load_const(self.index))
# TODO(anijain2305) - Change this to dict.__getitem__(d, k)
codegen.append_output(create_instruction("BINARY_SUBSCR"))
def name(self):

2
torch/_dynamo/utils.py
View file

@ -2409,7 +2409,7 @@ def to_subclass(t, cls):
def dict_keys_getitem(d, n):
return next(itertools.islice(iter(d), n, n + 1))
return next(itertools.islice(iter(dict(d)), n, n + 1))
def enum_repr(value, local):