From 8bd75e90cadbfbc408a71f0057033d15a93aa4bf Mon Sep 17 00:00:00 2001
From: Allison Piper <alliepiper16@gmail.com>
Date: Tue, 29 Oct 2024 15:24:37 +0000
Subject: [PATCH] Update matrix with cuda-ext configurations.

CCCL needs a couple of images with additional CTK libraries installed.
This adds images with cuBLAS, cuTensor, and cuSOLVER installed.
---
 matrix.yml | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/matrix.yml b/matrix.yml
index 2e513d5a..d2b5bab3 100644
--- a/matrix.yml
+++ b/matrix.yml
@@ -56,6 +56,17 @@ x-cccl-cuda-opts: &cccl_cuda_opts {
   installNVRTC: true,
 }
 
+x-cccl-cuda-ext-opts: &cccl_cuda_ext_opts {
+  installCTKLibraries: false,
+  installCUDARuntime: true,
+  installcuRAND: true,
+  installcuTensor: true,
+  installcuBLAS: true,
+  installcuSOLVER: true,
+  installNCCL: false,
+  installNVRTC: true,
+}
+
 include:
 # CCCL devcontainers
 
@@ -103,6 +114,7 @@ include:
   - { features: [*python, *gcc_12, { <<: *cuda_curr_max, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *gcc_env }
   - { features: [*python, *gcc_13, { <<: *cuda_curr_min, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *gcc_env }
   - { features: [*python, *gcc_13, { <<: *cuda_curr_max, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *gcc_env }
+  - { features: [*python, *gcc_13, { <<: *cuda_curr_max, <<: *cccl_cuda_ext_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *gcc_env }
   - { features: [*python, *llvm_15, { <<: *cuda_curr_min, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
   - { features: [*python, *llvm_15, { <<: *cuda_curr_max, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
   - { features: [*python, *llvm_16, { <<: *cuda_curr_min, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
@@ -113,6 +125,7 @@ include:
   - { features: [*python, *llvm_18, { <<: *cuda_curr_max, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
   - { features: [*python, *llvm_19, { <<: *cuda_curr_min, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
   - { features: [*python, *llvm_19, { <<: *cuda_curr_max, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
+  - { features: [*python, *llvm_19, { <<: *cuda_curr_max, <<: *cccl_cuda_ext_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *llvm_env }
   - { features: [*python, *oneapi_2022, { <<: *cuda_curr_max, <<: *cccl_cuda_opts }, *clang_format_cccl, *clangd_dev, *cccl_dev], env: *onapi_env }
 
 - os: "ubuntu:24.04"