ci: add docker-tags.yml to specify the docker image tag used in CI (#1853)

<!-- .github/pull_request_template.md -->

## 📌 Description

Followup of #1839 , this PR created a `docker-tags.yml` (placed under
`ci/`) file to specify the tag used for each docker image, format:
```
image: tag
```
Each tag is composed of date and git-hash.

For public/gitlab CI, we can read from this file to specify the CI
docker version.

The docker file update and unittest on new container are decoupled,
after we upgrade dockerfile, the ci will still be running in earlier tag
until we update the `docker-tags.yml`.

## 🔍 Related Issues

<!-- Link any related issues here -->

## 🚀 Pull Request Checklist

Thank you for contributing to FlashInfer! Before we review your pull
request, please make sure the following items are complete.

### ✅ Pre-commit Checks

- [x] I have installed `pre-commit` by running `pip install pre-commit`
(or used your preferred method).
- [x] I have installed the hooks with `pre-commit install`.
- [x] I have run the hooks manually with `pre-commit run --all-files`
and fixed any reported issues.

> If you are unsure about how to set up `pre-commit`, see [the
pre-commit documentation](https://pre-commit.com/).

## 🧪 Tests

- [x] Tests have been added or updated as needed.
- [x] All tests are passing (`unittest`, etc.).

## Reviewer Notes
cc @bkryu @nvmbreughe @yongwww

Author: yzh119

Jenkinsfile

@@ -37,13 +37,11 @@
 //
 
 import org.jenkinsci.plugins.pipeline.modeldefinition.Utils
-// These are set at runtime from data in ci/jenkins/docker-images.yml, update
-// image tags in that file
-// Now supports multiple CUDA versions
-docker_run_cu126 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu126:latest"
-docker_run_cu128 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu128:latest"
-docker_run_cu129 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu129:latest"
-docker_run_cu130 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu130:latest"
+
+def getDockerRun(cuda_version, dockerTags) {
+  def image_name = "flashinfer/flashinfer-ci-${cuda_version}"
+  return "bash ci/bash.sh ${image_name}:${dockerTags[image_name]}"
+}
 
 def per_exec_ws(folder) {
   return "workspace/exec_${env.EXECUTOR_NUMBER}/" + folder
@@ -181,19 +179,6 @@ def run_with_spot_retry(spot_node_type, on_demand_node_type, test_name, test_clo
 def run_unittest_CPU_JIT_CACHE_PACKAGE_BUILD_IMPORT(node_type, cuda_version) {
   echo "Running CPU JIT Cache Package Build and Import Unittest with CUDA ${cuda_version}"
 
-  def docker_run = ""
-  if (cuda_version == "cu126") {
-    docker_run = docker_run_cu126
-  } else if (cuda_version == "cu128") {
-    docker_run = docker_run_cu128
-  } else if (cuda_version == "cu129") {
-    docker_run = docker_run_cu129
-  } else if (cuda_version == "cu130") {
-    docker_run = docker_run_cu130
-  } else {
-    error("Unknown CUDA version: ${cuda_version}")
-  }
-
   if (node_type.contains('SPOT')) {
     // Add timeout only for spot instances - node allocation only
     def node_allocated = false
@@ -212,6 +197,8 @@ def run_unittest_CPU_JIT_CACHE_PACKAGE_BUILD_IMPORT(node_type, cuda_version) {
       node(node_type) {
         ws(per_exec_ws('flashinfer-jit-cache')) {
           init_git(true)
+          def dockerTags = readYaml file: 'ci/docker-tags.yml'
+          def docker_run = getDockerRun(cuda_version, dockerTags)
           sh(script: "ls -alh", label: 'Show work directory')
           sh(script: "./scripts/task_show_node_info.sh", label: 'Show node info')
           sh(script: "${docker_run} --no-gpu ./scripts/task_test_jit_cache_package_build_import.sh", label: 'Test JIT Cache Package Build and Import')
@@ -228,6 +215,8 @@ def run_unittest_CPU_JIT_CACHE_PACKAGE_BUILD_IMPORT(node_type, cuda_version) {
     node(node_type) {
       ws(per_exec_ws('flashinfer-jit-cache')) {
         init_git(true)
+        def dockerTags = readYaml file: 'ci/docker-tags.yml'
+        def docker_run = getDockerRun(cuda_version, dockerTags)
         sh(script: "ls -alh", label: 'Show work directory')
         sh(script: "./scripts/task_show_node_info.sh", label: 'Show node info')
         sh(script: "${docker_run} --no-gpu ./scripts/task_test_jit_cache_package_build_import.sh", label: 'Test JIT Cache Package Build and Import')
@@ -239,17 +228,6 @@ def run_unittest_CPU_JIT_CACHE_PACKAGE_BUILD_IMPORT(node_type, cuda_version) {
 def shard_run_unittest_GPU(node_type, shard_id, cuda_version) {
   echo "Running unittest on ${node_type}, shard ${shard_id}, CUDA ${cuda_version}"
 
-  def docker_run = ""
-  if (cuda_version == "cu126") {
-    docker_run = docker_run_cu126
-  } else if (cuda_version == "cu128") {
-    docker_run = docker_run_cu128
-  } else if (cuda_version == "cu129") {
-    docker_run = docker_run_cu129
-  } else {
-    error("Unknown CUDA version: ${cuda_version}")
-  }
-
   if (node_type.contains('SPOT')) {
     // Add timeout only for spot instances - node allocation only
     def node_allocated = false
@@ -268,6 +246,8 @@ def shard_run_unittest_GPU(node_type, shard_id, cuda_version) {
       node(node_type) {
         ws(per_exec_ws('flashinfer-unittest')) {
           init_git(true) // we need cutlass submodule
+          def dockerTags = readYaml file: 'ci/docker-tags.yml'
+          def docker_run = getDockerRun(cuda_version, dockerTags)
           sh(script: "ls -alh", label: 'Show work directory')
           sh(script: "./scripts/task_show_node_info.sh", label: 'Show node info')
           sh(script: "${docker_run} ./scripts/task_jit_run_tests_part${shard_id}.sh", label: 'JIT Unittest Part ${shard_id}')
@@ -284,6 +264,8 @@ def shard_run_unittest_GPU(node_type, shard_id, cuda_version) {
     node(node_type) {
       ws(per_exec_ws('flashinfer-unittest')) {
         init_git(true) // we need cutlass submodule
+        def dockerTags = readYaml file: 'ci/docker-tags.yml'
+        def docker_run = getDockerRun(cuda_version, dockerTags)
         sh(script: "ls -alh", label: 'Show work directory')
         sh(script: "./scripts/task_show_node_info.sh", label: 'Show node info')
         sh(script: "${docker_run} ./scripts/task_jit_run_tests_part${shard_id}.sh", label: 'JIT Unittest Part ${shard_id}')

ci/docker-tags.yml

@@ -0,0 +1,4 @@
+flashinfer/flashinfer-ci-cu126: 20251002-52089b5
+flashinfer/flashinfer-ci-cu128: 20251002-52089b5
+flashinfer/flashinfer-ci-cu129: 20251002-52089b5
+flashinfer/flashinfer-ci-cu130: 20251002-52089b5