From f55528a4792aaccaade36fd47b1ae66c07e7ac0f Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Mon, 21 Oct 2024 13:45:44 -0700
Subject: [PATCH] Add support for Ubuntu 24.04.

---
 ci/compute-matrix.jq | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ci/compute-matrix.jq b/ci/compute-matrix.jq
index cae9d3fb..32064dcf 100644
--- a/ci/compute-matrix.jq
+++ b/ci/compute-matrix.jq
@@ -3,6 +3,10 @@ def compute_arch($x):
 
 def compute_ubuntu_version($x):
   if
+    $x.CUDA_VER >= "12.5" # Ubuntu 24.04 nvidia/cuda images were added starting with CUDA 12.5
+  then
+    ["ubuntu", "24.04"]
+  elif
     $x.CUDA_VER >= "11.7" # Ubuntu 22.04 nvidia/cuda images were added starting with CUDA 11.7
   then
     ["ubuntu", "22.04"]