chore: add ucx binaries to path (#3388)

Signed-off-by: alec-flowers <aflowers@nvidia.com>

container/Dockerfile

@@ -205,7 +205,7 @@ RUN --mount=type=secret,id=aws-key-id,env=AWS_ACCESS_KEY_ID \
 
 # UCX environment variables
 ENV CPATH=/usr/include:$CPATH \
-    PATH=/usr/bin:$PATH \
+    PATH=/usr/bin:/usr/local/ucx/bin:$PATH \
     PKG_CONFIG_PATH=/usr/lib/pkgconfig:$PKG_CONFIG_PATH
 
 ##################################

container/Dockerfile.trtllm

@@ -116,12 +116,16 @@ COPY --from=framework /usr/local/lib/lib* /usr/local/lib/
 # Copy nats and etcd from dynamo_base image
 COPY --from=dynamo_base /usr/bin/nats-server /usr/bin/nats-server
 COPY --from=dynamo_base /usr/local/bin/etcd/ /usr/local/bin/etcd/
+# Add ETCD and CUDA binaries to PATH so cicc and other CUDA tools are accessible
+ENV PATH=/usr/local/bin/etcd/:/usr/local/cuda/nvvm/bin:$PATH
 
 # Copy UCX from framework image as plugin for NIXL
 # Copy NIXL source from framework image
 # Copy dynamo wheels for gitlab artifacts
 COPY --from=dynamo_base /usr/local/ucx /usr/local/ucx
 COPY --from=dynamo_base $NIXL_PREFIX $NIXL_PREFIX
+ENV PATH=/usr/local/ucx/bin:$PATH
+
 # Copy OpenMPI from framework image
 COPY --from=framework /opt/hpcx/ompi /opt/hpcx/ompi
 # Copy NUMA library from framework image
@@ -339,4 +343,4 @@ RUN uv pip install maturin[patchelf]
 COPY pyproject.toml README.md hatch_build.py /workspace/
 RUN uv pip install --no-deps -e .
 
-CMD []
\ No newline at end of file
+CMD []

container/Dockerfile.vllm

@@ -221,6 +221,7 @@ ENV PATH=/usr/local/bin/etcd/:/usr/local/cuda/nvvm/bin:$PATH
 # Copy dynamo wheels for gitlab artifacts
 COPY --from=dynamo_base /usr/local/ucx /usr/local/ucx
 COPY --from=dynamo_base $NIXL_PREFIX $NIXL_PREFIX
+ENV PATH=/usr/local/ucx/bin:$PATH
 
 # Copies vllm, DeepEP, DeepGEMM, PPLX repos (all editable installs) and nvshmem binaries
 COPY --from=framework /opt/vllm /opt/vllm