commit 702880e66fc6a62d3c00f2b11ddb87d041773731 · pyrox.dev/nixpkgs

+7 -3

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

···

       181
        
           else if cudaSupport then

     

       182
        
             gpuArchWarner supportedCudaCapabilities unsupportedCudaCapabilities

     

       183
        
           else if rocmSupport then

     

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

     

       185
       -
             # use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'

     

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

     

       187
        
             lib.lists.subtractLists [

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       188
        
               "gfx1010"

     

       189
        
               "gfx1012"

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

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

     

       191
        
           else

     

       192
        
             throw "No GPU targets specified"

···

       181
        
           else if cudaSupport then

     

       182
        
             gpuArchWarner supportedCudaCapabilities unsupportedCudaCapabilities

     

       183
        
           else if rocmSupport then

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       184
        
             lib.lists.subtractLists [

     

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

     

       186
       +
               # use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'

     

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

     

       188
        
               "gfx1010"

     

       189
        
               "gfx1012"

     

       190
       +
       

     

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

     

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

     

       193
       +
               "gfx1151"

     

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

     

       195
        
           else

     

       196
        
             throw "No GPU targets specified"

+1

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

···

       236
        
             "1100"

     

       237
        
             "1101"

     

       238
        
             "1102"

     

       0
        
       
     

       239
        
             "1200" # RX 9070

     

       240
        
             "1201" # RX 9070 XT

     

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

···

       236
        
             "1100"

     

       237
        
             "1101"

     

       238
        
             "1102"

     

       239
       +
             "1151" # Strix Halo

     

       240
        
             "1200" # RX 9070

     

       241
        
             "1201" # RX 9070 XT

     

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

+1

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

···

       498
        
           "gfx1100"

     

       499
        
           "gfx1101"

     

       500
        
           "gfx1102"

     

       0
        
       
     

       501
        
         ];

     

       502
        
         gfx12 = scopeForArches [

     

       503
        
           "gfx1200"

···

       498
        
           "gfx1100"

     

       499
        
           "gfx1101"

     

       500
        
           "gfx1102"

     

       501
       +
           "gfx1151"

     

       502
        
         ];

     

       503
        
         gfx12 = scopeForArches [

     

       504
        
           "gfx1200"

+1

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

+1

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