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

[TD] Enable TD on distributed cpu (#150028)

Browse Source 
Enable TD on distributed cpu, I think the only reason it's not is because I forgot to enable it

Get rid of some of the statements that are no ops:
* asan uses default shard
* nogpu got moved to periodic
* no windows cuda testing anymore

Only thing on pull and trunk that doesn't use TD is dynamo_wrapped but I think it's fast enough to be ok for now, we can take another look after this
Pull Request resolved: https://github.com/pytorch/pytorch/pull/150028
Approved by: https://github.com/ZainRizvi
This commit is contained in:
Catherine Lee
2025-03-28 17:19:11 +00:00
committed by PyTorch MergeBot
parent cf7447ae99
commit 85079e4380
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/run_test.py
View File

@ -30,7 +30,6 @@ from torch.testing._internal.common_utils import (
get_report_path,
IS_CI,
IS_MACOS,
IS_WINDOWS,
retry_shell,
set_cwd,
shell,
@ -1377,11 +1376,7 @@ def parse_args():
default=IS_CI
and (
TEST_WITH_CROSSREF
or TEST_WITH_ASAN
or (TEST_CONFIG == "distributed" and TEST_CUDA)
or (IS_WINDOWS and not TEST_CUDA)
or TEST_CONFIG == "nogpu_AVX512"
or TEST_CONFIG == "nogpu_NO_AVX2"
or TEST_CONFIG == "distributed"
or TEST_CONFIG == "default"
)
and get_pr_number() is not None