[misc] Use transient-package for triton installation

airootfs/root/customize_airootfs.sh.jinja2

@@ -34,6 +34,9 @@ systemctl enable nvidia-persistenced
 systemctl enable ly
 {% endif %}
 
+# install transient-package
+pipx install transient-package
+
 {% if type != "empty" %}
 # enter home directory
 cd "/home/tori"

airootfs/root/customize_airootfs/scripts/1000-vllm.sh.jinja2

@@ -7,10 +7,4 @@ export PIP_EXTRA_INDEX_URL="https://sasha0552.github.io/pascal-pkgs-ci/"
 
 # install patched vllm
 pipx install vllm-pascal
-
-# remove triton
-pipx runpip vllm-pascal uninstall -y triton
-
-# install patched triton
-pipx runpip vllm-pascal install triton-pascal
 {% endif %}

airootfs/root/customize_airootfs/scripts/2000-replace-triton.sh.jinja2

@@ -0,0 +1,20 @@
+#!/bin/sh
+set -eu
+
+{% if platform == "cuda" %}
+# use repo with patched packages
+export PIP_EXTRA_INDEX_URL="https://sasha0552.github.io/pascal-pkgs-ci/"
+
+# Iterate over all virtual environments with triton
+for dir in $(find /home/tori -type d -wholename "*/site-packages/triton"); do
+  # Activate virtual environment
+  source "$dir/../../../../bin/activate"
+
+  # Replace triton
+  transient-package install \
+    --source triton         \
+    --target triton-pascal
+
+  # Deactivate virtual environment
+  deactivate
+{% endif %}