Allow building specific variants

Author: cmdr2

scripts/build/__main__.py

@@ -18,10 +18,16 @@ def main():
         choices=["x64", "arm64"],
         help="The architecture to build for (default: detected from current platform)",
     )
+    parser.add_argument(
+        "--variant",
+        default=None,
+        help="The variant to build for (e.g., 'sm86' for CUDA). If not specified, the default variant for the platform will be used.",
+    )
     args = parser.parse_args()
 
     platform = args.platform
     arch = args.arch
+    variant = args.variant
     try:
         module = globals()[f"platform_{platform}"]
     except KeyError:
@@ -45,6 +51,7 @@ def main():
         additional_files_func,
         env_func,
         arch,
+        variant,
     )
 
 

scripts/build/common.py

@@ -188,6 +188,7 @@ def build_project(
     get_additional_files_func=None,
     get_env_func=None,
     arch=None,
+    variant=None,
 ):
     """Common build logic for all backends.
 
@@ -210,6 +211,13 @@ def build_project(
     if not variants:
         variants = [{"name": "any", "compile_flags": []}]
 
+    if variant is not None:
+        # Filter variants to only include the specified one
+        variants = [v for v in variants if v["name"] == variant]
+        if not variants:
+            print(f"Variant '{variant}' not found.")
+            sys.exit(1)
+
     # Get target for "any" variant to use for additional files
     target_any = get_target(get_platform_name_func, "any", arch)