commit 64169e994ab641838b9ca76f9a642613f674bd90 · pyrox.dev/nixpkgs

-9

pkgs/development/cuda-modules/packages/saxpy/package.nix

···

       43
       43
        
         cmakeFlags = [

     

       44
       44
        
           (lib.cmakeBool "CMAKE_VERBOSE_MAKEFILE" true)

     

       45
       45
        
           (lib.cmakeFeature "CMAKE_CUDA_ARCHITECTURES" flags.cmakeCudaArchitecturesString)

     

       46
       46
       -
           (lib.cmakeBool "CMAKE_SKIP_INSTALL_RPATH" false)

     

       47
       47
       -
           (lib.cmakeBool "CMAKE_SKIP_RPATH" false)

     

       48
       48
       -
           (lib.cmakeBool "CMAKE_INSTALL_RPATH_USE_LINK_PATH" true)

     

       49
       49
       -
           (lib.cmakeFeature "CMAKE_INSTALL_RPATH" (

     

       50
       50
       -
             lib.makeLibraryPath [

     

       51
       51
       -
               (getLib libcublas)

     

       52
       52
       -
               cuda_cudart

     

       53
       53
       -
             ]

     

       54
       54
       -
           ))

     

       55
       46
        
         ];

     

       56
       47
        
       

     

       57
       48
        
         passthru.gpuCheck = saxpy.overrideAttrs (_: {

+7 -2

pkgs/development/cuda-modules/packages/setupCudaHook/setup-cuda-hook.sh

···

       54
       54
        
               fi

     

       55
       55
        
           done

     

       56
       56
        
       

     

       57
       57
       -
           appendToVar cmakeFlags "-DCUDAToolkit_INCLUDE_DIR=$CUDAToolkit_INCLUDE_DIR"

     

       58
       58
       -
           appendToVar cmakeFlags "-DCUDAToolkit_ROOT=$CUDAToolkit_ROOT"

     

       57
       57
       +
           # Use array form so semicolon-separated lists are passed safely.

     

       58
       58
       +
           if [[ -n "${CUDAToolkit_INCLUDE_DIR-}" ]]; then

     

       59
       59
       +
               cmakeFlagsArray+=("-DCUDAToolkit_INCLUDE_DIR=${CUDAToolkit_INCLUDE_DIR}")

     

       60
       60
       +
           fi

     

       61
       61
       +
           if [[ -n "${CUDAToolkit_ROOT-}" ]]; then

     

       62
       62
       +
               cmakeFlagsArray+=("-DCUDAToolkit_ROOT=${CUDAToolkit_ROOT}")

     

       63
       63
       +
           fi

     

       59
       64
        
       }

     

       60
       65
        
       preConfigureHooks+=(setupCUDAToolkit_ROOT)

     

       61
       66