fix: Install specific toolchain (#329)

`cargo build --locked` won't let you use "1.85.0" if you only have "stable" installed, even if those are the same thing right now.

.github/workflows/pre-merge-rust.yml

@@ -66,6 +66,7 @@ jobs:
       if: ${{ env.ACT }}
       run: |
         curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+        rustup toolchain install 1.85.0-x86_64-unknown-linux-gnu
         echo "$HOME/.cargo/bin" >> $GITHUB_PATH
     - name: Set up Rust Toolchain Components
       run: rustup component add rustfmt clippy

container/Dockerfile

@@ -30,6 +30,7 @@ RUN apt-get update && \
 
 RUN curl https://sh.rustup.rs -sSf | bash -s -- -y
 ENV PATH="/root/.cargo/bin:${PATH}"
+RUN rustup toolchain install 1.85.0-x86_64-unknown-linux-gnu
 
 # Install OpenAI-compatible frontend and its dependencies from triton server
 # repository. These are used to have a consistent interface, schema, and FastAPI

container/Dockerfile.vllm

@@ -69,6 +69,7 @@ RUN apt update -y && \
     pkg-config && \
     curl https://sh.rustup.rs -sSf | bash -s -- -y
 ENV PATH="/root/.cargo/bin:${PATH}"
+RUN rustup toolchain install 1.85.0-x86_64-unknown-linux-gnu
 
 # Working directory
 WORKDIR /workspace
@@ -168,4 +169,4 @@ WORKDIR /workspace
 # FIXME: May want a modification with triton-distributed banner on entry
 ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
 
-CMD []
\ No newline at end of file
+CMD []

container/Dockerfile.vllm_nixl

@@ -199,6 +199,7 @@ RUN apt update -y && \
     pkg-config && \
     curl https://sh.rustup.rs -sSf | bash -s -- -y
 ENV PATH="/root/.cargo/bin:${PATH}"
+RUN rustup toolchain install 1.85.0-x86_64-unknown-linux-gnu
 
 # Working directory
 WORKDIR /workspace

launch/tio/rust-toolchain.toml

 [toolchain]
-channel = "1.84.1"
+channel = "1.85.0"

lib/bindings/python/rust-toolchain.toml

 [toolchain]
-channel = "1.84.1"
+channel = "1.85.0"

lib/llm/rust-toolchain.toml

 [toolchain]
-channel = "1.84.1"
+channel = "1.85.0"

lib/runtime/rust-toolchain.toml

 [toolchain]
-channel = "1.84.1"
+channel = "1.85.0"