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

Bump STATIC_CUDA_LAUNCHER_VERSION to 2 (#156726)

Browse Source 
Differential Revision: [D77241813](https://our.internmc.facebook.com/intern/diff/D77241813)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156726
Approved by: https://github.com/oulgen
This commit is contained in:
James Wu
2025-06-24 10:55:47 -07:00
committed by PyTorch MergeBot
parent b5bfbba184
commit e581f015ee
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_inductor/config.py
View File

@ -40,7 +40,7 @@ def bundle_triton_into_fx_graph_cache_default() -> Optional[bool]:
def static_cuda_launcher_default() -> bool: def static_cuda_launcher_default() -> bool:
STATIC_CUDA_LAUNCHER_VERSION = 1 STATIC_CUDA_LAUNCHER_VERSION = 2
if "TORCHINDUCTOR_USE_STATIC_CUDA_LAUNCHER" in os.environ: if "TORCHINDUCTOR_USE_STATIC_CUDA_LAUNCHER" in os.environ:
return os.environ.get("TORCHINDUCTOR_USE_STATIC_CUDA_LAUNCHER") == "1" return os.environ.get("TORCHINDUCTOR_USE_STATIC_CUDA_LAUNCHER") == "1"