Skip to content

Commit 7c2586e

Browse files
committed
Set explicit dependencies
1 parent 98d5574 commit 7c2586e

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

build.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1322,12 +1322,12 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
13221322
if target_machine == "aarch64":
13231323
backend_dependencies += " libgfortran5"
13241324

1325-
if "CUDA_VERSION" in os.environ:
1326-
cuda_version_major = int(os.environ["CUDA_VERSION"].split(".")[0])
1327-
if cuda_version_major >= 13:
1328-
backend_dependencies += " libnvshmem3-cuda-{}".format(
1329-
cuda_version_major
1330-
)
1325+
if (
1326+
target_platform != "igpu"
1327+
or target_platform != "windows"
1328+
or target_platform != "rhel"
1329+
):
1330+
backend_dependencies += " libnvshmem3-cuda-13"
13311331

13321332
# openssh-server is needed for fastertransformer
13331333
if "fastertransformer" in backends:

0 commit comments

Comments
 (0)