frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-13 21:59:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Only use clang/lld for CUDA

Browse Source
This commit is contained in:
Bert Maher
2024-11-15 08:56:37 -08:00
parent d771851d87
commit fe97150a81
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.github/scripts/build_triton_wheel.py vendored
View File

@ -133,7 +133,8 @@ def build_triton(
# change built wheel name and version
env["TRITON_WHEEL_NAME"] = triton_pkg_name
env["TRITON_BUILD_WITH_CLANG_LLD"] = "1"
if device == "cuda":
env["TRITON_BUILD_WITH_CLANG_LLD"] = "1"
patch_init_py(
triton_pythondir / "triton" / "__init__.py",
version=f"{version}",