commit 702880e66fc6a62d3c00f2b11ddb87d041773731 · pyrox.dev/nixpkgs

+7 -3

pkgs/development/python-modules/torch/source/default.nix

···

       181
       181
        
           else if cudaSupport then

     

       182
       182
        
             gpuArchWarner supportedCudaCapabilities unsupportedCudaCapabilities

     

       183
       183
        
           else if rocmSupport then

     

       184
       184
       -
             # Remove RDNA1 gfx101x archs from default ROCm support list to avoid

     

       185
       185
       -
             # use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'

     

       186
       186
       -
             # TODO: Retest after ROCm 6.4 or torch 2.8

     

       187
       184
        
             lib.lists.subtractLists [

     

       185
       185
       +
               # Remove RDNA1 gfx101x archs from default ROCm support list to avoid

     

       186
       186
       +
               # use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'

     

       187
       187
       +
               # TODO: Retest after ROCm 6.4 or torch 2.8

     

       188
       188
        
               "gfx1010"

     

       189
       189
        
               "gfx1012"

     

       190
       190
       +
       

     

       191
       191
       +
               # Strix Halo seems to be broken as well, see

     

       192
       192
       +
               # https://github.com/NixOS/nixpkgs/pull/440359.

     

       193
       193
       +
               "gfx1151"

     

       190
       194
        
             ] (rocmPackages.clr.localGpuTargets or rocmPackages.clr.gpuTargets)

     

       191
       195
        
           else

     

       192
       196
        
             throw "No GPU targets specified"

+1

pkgs/development/rocm-modules/6/clr/default.nix

···

       236
       236
        
             "1100"

     

       237
       237
        
             "1101"

     

       238
       238
        
             "1102"

     

       239
       239
       +
             "1151" # Strix Halo

     

       239
       240
        
             "1200" # RX 9070

     

       240
       241
        
             "1201" # RX 9070 XT

     

       241
       242
        
           ] (target: "gfx${target}");

+1

pkgs/development/rocm-modules/6/default.nix

···

       498
       498
        
           "gfx1100"

     

       499
       499
        
           "gfx1101"

     

       500
       500
        
           "gfx1102"

     

       501
       501
       +
           "gfx1151"

     

       501
       502
        
         ];

     

       502
       503
        
         gfx12 = scopeForArches [

     

       503
       504
        
           "gfx1200"

+1

pkgs/development/rocm-modules/6/rocblas/default.nix

+1

pkgs/development/rocm-modules/6/rocsolver/default.nix