···

       
43
       
 
       
  cmakeFlags = [

     

       
44
       
 
       
    (lib.cmakeBool "CMAKE_VERBOSE_MAKEFILE" true)

     

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

     

       
46
       
-
       
    (lib.cmakeBool "CMAKE_SKIP_INSTALL_RPATH" false)

     

       
47
       
-
       
    (lib.cmakeBool "CMAKE_SKIP_RPATH" false)

     

       
48
       
-
       
    (lib.cmakeBool "CMAKE_INSTALL_RPATH_USE_LINK_PATH" true)

     

       
49
       
-
       
    (lib.cmakeFeature "CMAKE_INSTALL_RPATH" (

     

       
50
       
-
       
      lib.makeLibraryPath [

     

       
51
       
-
       
        (getLib libcublas)

     

       
52
       
-
       
        cuda_cudart

     

       
53
       
-
       
      ]

     

       
54
       
-
       
    ))

     

       
55
       
 
       
  ];

     

       
56
       
 
       


     

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

     

···

       
43
       
 
       
  cmakeFlags = [

     

       
44
       
 
       
    (lib.cmakeBool "CMAKE_VERBOSE_MAKEFILE" true)

     

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

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
46
       
 
       
  ];

     

       
47
       
 
       


     

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

     

···

       
54
       
 
       
        fi

     

       
55
       
 
       
    done

     

       
56
       
 
       


     

       
57
       
-
       
    appendToVar cmakeFlags "-DCUDAToolkit_INCLUDE_DIR=$CUDAToolkit_INCLUDE_DIR"

     

       
58
       
-
       
    appendToVar cmakeFlags "-DCUDAToolkit_ROOT=$CUDAToolkit_ROOT"

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
59
       
 
       
}

     

       
60
       
 
       
preConfigureHooks+=(setupCUDAToolkit_ROOT)

     

       
61
       
 
       


     

···

       
54
       
 
       
        fi

     

       
55
       
 
       
    done

     

       
56
       
 
       


     

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

     

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

     

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

     

       
60
       
+
       
    fi

     

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

     

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

     

       
63
       
+
       
    fi

     

       
64
       
 
       
}

     

       
65
       
 
       
preConfigureHooks+=(setupCUDAToolkit_ROOT)

     

       
66