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 update)

Browse source 
[ghstack-poisoned]
This commit is contained in:
Xuehai Pan 2025-01-10 01:12:17 +08:00
parent db08e49d79
commit ac7fd643ff
3 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.lintrunner.toml
View file

@ -171,7 +171,7 @@ include_patterns = [
'benchmarks/instruction_counts/**/*.py',
'tools/**/*.py',
'torchgen/**/*.py',
'torch/utils/pytree.py',
'torch/utils/pytree/__init__.py',
'torch/utils/_pytree.py',
'torch/utils/_cxx_pytree.py',
'torch/utils/benchmark/utils/common.py',

2
mypy-strict.ini
View file

@ -29,7 +29,7 @@ files =
benchmarks/instruction_counts,
tools,
torch/profiler/_memory_profiler.py,
torch/utils/pytree.py,
torch/utils/pytree/__init__.py,
torch/utils/_pytree.py,
torch/utils/_cxx_pytree.py,
torch/utils/benchmark/utils/common.py,

15
torch/utils/pytree/__init__.py
View file

@ -77,14 +77,19 @@ __all__ = [
"tree_all_only",
"tree_any_only",
"treespec_pprint",
"FlattenFunc",
"UnflattenFunc",
"FlattenWithKeysFunc",
"ToDumpableContextFunc",
"FromDumpableContextFunc",
]
if _TYPE_CHECKING:
__all__ += [
"FlattenFunc",
"UnflattenFunc",
"FlattenWithKeysFunc",
"ToDumpableContextFunc",
"FromDumpableContextFunc",
]
PYTORCH_USE_CXX_PYTREE: bool = _os.getenv("PYTORCH_USE_CXX_PYTREE", "0") not in {
"0",
"",