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

[2/N] More ruff SIM fixes (#165031)

Browse Source 
This is follow-up of #164695 to apply ruff SIM rules to more files. Most changes are about simplifying dict.get because None is already the default value.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165031
Approved by: https://github.com/mlazos
This commit is contained in:
Yuanyuan Chen
2025-10-14 14:22:49 +00:00
committed by PyTorch MergeBot
parent 1fa11f42b1
commit fbe0d20a17
52 changed files with 98 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
torch/_inductor/codegen/common.py
View File

@ -482,15 +482,11 @@ def get_wrapper_codegen_for_device(
def get_custom_backend_pass_for_device(device: str) -> Optional[CustomGraphModulePass]:
return custom_backend_passes[device] if device in custom_backend_passes else None
return custom_backend_passes.get(device)
def get_custom_backend_config_for_device(device: str) -> Optional[ConfigModule]:
return (
custom_backend_codegen_configs[device]
if device in custom_backend_codegen_configs
else None
)
return custom_backend_codegen_configs.get(device)
@functools.cache