[misc] Use transient-package for triton installation

airootfs/root/customize_airootfs/scripts/0000-transient-package.sh.jinja2

@@ -0,0 +1,5 @@
+#!/bin/sh
+set -eu
+
+# install transient-package
+pipx install transient-package

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,17 @@
+#!/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
+{% endif %}