@@ -655,23 +655,47 @@ def create_BCI(
655655
656656NVIDIA_CONTAINERS = [
657657 create_BCI (
658- build_tag = f"third-party/nvidia/driver:{ driver_ver } -sles{ os_ver } " ,
658+ build_tag = (
659+ f"third-party/nvidia/driver:{ driver_ver } -sles{ os_ver } "
660+ if kernel_flavor == "default"
661+ else f"third-party/nvidia/driver:{ driver_ver } -{ kernel_flavor } -sles{ os_ver } "
662+ ),
659663 available_versions = [f"{ os_ver } -third-party" ],
660664 custom_entry_point = "/bin/sh" ,
661- )
662- for driver_ver , os_ver in (
663- ("595.45.04" , "16.0" ),
664- ("595.45.04" , "15.7" ),
665- ("590.48.01" , "15.7" ),
666- ("580.126.16" , "15.7" ),
667- ("580.126.09" , "15.7" ),
668- ("580.105.08" , "15.7" ),
669- ("580.95.05" , "15.7" ),
670- ("580.82.07" , "15.7" ),
671- ("575.57.08" , "15.7" ),
672- ("570.211.01" , "15.7" ),
673- ("570.195.03" , "15.7" ),
674- ("550.163.01" , "15.7" ),
665+ extra_marks = [
666+ pytest .mark .skipif (
667+ condition = LOCALHOST .system_info .arch != "aarch64" ,
668+ reason = "64kb flavor is available only on aarch64" ,
669+ )
670+ ]
671+ if kernel_flavor == "64kb"
672+ else [],
673+ )
674+ for driver_ver , kernel_flavor , os_ver in (
675+ ("595.45.04" , "default" , "16.0" ),
676+ ("595.45.04" , "64kb" , "16.0" ),
677+ ("595.45.04" , "default" , "15.7" ),
678+ ("595.45.04" , "64kb" , "15.7" ),
679+ ("590.48.01" , "default" , "15.7" ),
680+ ("590.48.01" , "64kb" , "15.7" ),
681+ ("580.126.16" , "default" , "15.7" ),
682+ ("580.126.16" , "64kb" , "15.7" ),
683+ ("580.126.09" , "default" , "15.7" ),
684+ ("580.126.09" , "64kb" , "15.7" ),
685+ ("580.105.08" , "default" , "15.7" ),
686+ ("580.105.08" , "64kb" , "15.7" ),
687+ ("580.95.05" , "default" , "15.7" ),
688+ ("580.95.05" , "64kb" , "15.7" ),
689+ ("580.82.07" , "default" , "15.7" ),
690+ ("580.82.07" , "64kb" , "15.7" ),
691+ ("575.57.08" , "default" , "15.7" ),
692+ ("575.57.08" , "64kb" , "15.7" ),
693+ ("570.211.01" , "default" , "15.7" ),
694+ ("570.211.01" , "64kb" , "15.7" ),
695+ ("570.195.03" , "default" , "15.7" ),
696+ ("570.195.03" , "64kb" , "15.7" ),
697+ ("550.163.01" , "default" , "15.7" ),
698+ ("550.163.01" , "64kb" , "15.7" ),
675699 )
676700]
677701
0 commit comments