cuda-modules: fix sort predicate stability (#368366)
This commit is contained in:
commit
2cd2decd3a
@ -81,7 +81,7 @@ let
|
|||||||
redistArch = flags.getRedistArch hostPlatform.system;
|
redistArch = flags.getRedistArch hostPlatform.system;
|
||||||
|
|
||||||
preferable =
|
preferable =
|
||||||
p1: p2: (isSupported p2 -> isSupported p1) && (strings.versionAtLeast p1.version p2.version);
|
p1: p2: (isSupported p2 -> isSupported p1) && (strings.versionOlder p2.version p1.version);
|
||||||
|
|
||||||
# All the supported packages we can build for our platform.
|
# All the supported packages we can build for our platform.
|
||||||
# perSystemReleases :: List Package
|
# perSystemReleases :: List Package
|
||||||
|
|||||||
@ -364,7 +364,7 @@ buildPythonPackage rec {
|
|||||||
''
|
''
|
||||||
+ lib.optionalString (cudaSupport && cudaPackages ? cudnn) ''
|
+ lib.optionalString (cudaSupport && cudaPackages ? cudnn) ''
|
||||||
export CUDNN_INCLUDE_DIR=${lib.getLib cudnn}/include
|
export CUDNN_INCLUDE_DIR=${lib.getLib cudnn}/include
|
||||||
export CUDNN_LIB_DIR=${cudnn.lib}/lib
|
export CUDNN_LIB_DIR=${lib.getLib cudnn}/lib
|
||||||
''
|
''
|
||||||
+ lib.optionalString rocmSupport ''
|
+ lib.optionalString rocmSupport ''
|
||||||
export ROCM_PATH=${rocmtoolkit_joined}
|
export ROCM_PATH=${rocmtoolkit_joined}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user