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

Browse source 
[ghstack-poisoned]
This commit is contained in:
Xuehai Pan 2025-02-09 22:38:06 +08:00
commit 5b88785d83
3 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
test/test_fx.py
View file

@ -32,10 +32,7 @@ from fx.test_fx_const_fold import TestConstFold # noqa: F401
from fx.test_fx_param_shape_control_flow import ( # noqa: F401
TestConstParamShapeInControlFlow,
)
from fx.test_gradual_type import ( # noqa: F401 # noqa: F401
AnnotationsTest,
TypeCheckerTest,
)
from fx.test_gradual_type import AnnotationsTest, TypeCheckerTest # noqa: F401
from fx.test_matcher_utils import TestMatcher # noqa: F401
from fx.test_pass_infra import TestPassManager # noqa: F401
from fx.test_source_matcher_utils import TestSourceMatcher # noqa: F401

2
torch/utils/_cxx_pytree.py
View file

@ -992,6 +992,8 @@ def treespec_loads(serialized: str) -> TreeSpec:
class _Asterisk(str):
__slots__ = ()
def __new__(cls) -> Self:
return super().__new__(cls, "*")

2
torch/utils/_pytree.py
View file

@ -1634,6 +1634,8 @@ def treespec_loads(serialized: str) -> TreeSpec:
class _Asterisk(str):
__slots__ = ()
def __new__(cls) -> Self:
return super().__new__(cls, "*")