···

       
150
       
150
       
 
       
  supportedCudaCapabilities = lists.intersectLists cudaFlags.cudaCapabilities supportedTorchCudaCapabilities;

     

       
151
       
151
       
 
       
  unsupportedCudaCapabilities = lists.subtractLists supportedCudaCapabilities cudaFlags.cudaCapabilities;

     

       
152
       
152
       
 
       


     

       
153
       
153
       
+
       
  isCudaJetson = cudaSupport && cudaPackages.cudaFlags.isJetsonBuild;

     

       
154
       
154
       
+
       


     

       
153
       
155
       
 
       
  # Use trivial.warnIf to print a warning if any unsupported GPU targets are specified.

     

       
154
       
156
       
 
       
  gpuArchWarner =

     

       
155
       
157
       
 
       
    supported: unsupported:

     
···

       
456
       
458
       
 
       
        cuda_nvcc

     

       
457
       
459
       
 
       
      ]

     

       
458
       
460
       
 
       
    )

     

       
461
       
461
       
+
       
    ++ lib.optionals isCudaJetson [ cudaPackages.autoAddCudaCompatRunpath ]

     

       
459
       
462
       
 
       
    ++ lib.optionals rocmSupport [ rocmtoolkit_joined ];

     

       
460
       
463
       
 
       


     

       
461
       
464
       
 
       
  buildInputs =