[Easy] Consolidate string startswith checks

[mrshenli]

Stack from ghstack (oldest at bottom):

• [Easy] Include SPMD and DTensor files in UFMT checks #98148
• -> [Easy] Consolidate string startswith checks #98147
• [SPMD] Enable fused Adam in full train step tracing #98113
• Format DTensor dispatch.py and _meta_registrations.py #98114

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/98147

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 Failures

As of commit 45fc422:

NEW FAILURES - The following jobs have failed:

• linux-focal-py3.9-clang7-asan / test (default, 1, 6, linux.4xlarge) (gh)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[mrshenli]

Test failure is irrelevant

test_jit_fuser_te.py::TestTEFuserDynamic::test_cat_2k_args Command took >20min, returning 124
FINISHED PRINTING LOG FILE of test_jit_fuser_te 1/3 (/var/lib/jenkins/workspace/test/test-reports/test_jit_fuser_te_1cy2lsh9.log)

test_jit_fuser_te 1/3 failed!
Traceback (most recent call last):
  File "/var/lib/jenkins/workspace/test/run_test.py", line 1398, in <module>
    main()
  File "/var/lib/jenkins/workspace/test/run_test.py", line 1356, in main
    raise RuntimeError(
RuntimeError: test_jit_fuser_te 1/3 failed!

[fegin]

LGTM

[mrshenli]

@pytorchbot merge -f "test error is irrelevant"

[pytorchmergebot]

Merge started

Your change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging

Check the merge workflow status
here