From d43affbd43c33ac2e27a3a9bfe693bc96373a34e Mon Sep 17 00:00:00 2001
From: Connor Baker <ConnorBaker01@gmail.com>
Date: Tue, 10 Jun 2025 20:07:15 +0000
Subject: [PATCH] pkgsCuda: init

Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
---
 pkgs/top-level/variants.nix | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pkgs/top-level/variants.nix b/pkgs/top-level/variants.nix
index ad4748f8b782..3d6b4bd44e99 100644
--- a/pkgs/top-level/variants.nix
+++ b/pkgs/top-level/variants.nix
@@ -95,6 +95,15 @@ self: super: {
     };
   });
 
+  # Full package set with cuda on rocm off
+  # Mostly useful for asserting pkgs.pkgsCuda.torchWithCuda == pkgs.torchWithCuda and similar
+  pkgsCuda = nixpkgsFun {
+    config = super.config // {
+      cudaSupport = true;
+      rocmSupport = false;
+    };
+  };
+
   # `pkgsForCudaArch` maps each CUDA capability in _cuda.db.cudaCapabilityToInfo to a Nixpkgs variant configured for
   # that target system. For example, `pkgsForCudaArch.sm_90a.python3Packages.torch` refers to PyTorch built for the
   # Hopper architecture, leveraging architecture-specific features.