23b612e36f
GitOrigin-RevId: ae5c332cbb5827f6b1f02572496b141021de335f
14 lines
416 B
Nix
14 lines
416 B
Nix
{hostPlatform, lib}:
|
|
let
|
|
# Samples are built around the CUDA Toolkit, which is not available for
|
|
# aarch64. Check for both CUDA version and platform.
|
|
platformIsSupported = hostPlatform.isx86_64 && hostPlatform.isLinux;
|
|
|
|
# Build our extension
|
|
extension =
|
|
final: _:
|
|
lib.attrsets.optionalAttrs platformIsSupported {
|
|
cuda-library-samples = final.callPackage ./generic.nix {};
|
|
};
|
|
in
|
|
extension
|