mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Bump STATIC_CUDA_LAUNCHER_VERSION to 2 (#156726)
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:
committed by
PyTorch MergeBot
parent
b5bfbba184
commit
e581f015ee
@ -40,7 +40,7 @@ def bundle_triton_into_fx_graph_cache_default() -> Optional[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:
|
||||
return os.environ.get("TORCHINDUCTOR_USE_STATIC_CUDA_LAUNCHER") == "1"
|
||||
|
Reference in New Issue
Block a user