2024-01-02 11:29:13 +00:00
|
|
|
{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;
|
|
|
|
|
|
|
|
|
|
# Build our extension
|
|
|
|
|
extension =
|
|
|
|
|
final: _:
|
|
|
|
|
lib.attrsets.optionalAttrs platformIsSupported {
|
|
|
|
|
cuda-library-samples = final.callPackage ./generic.nix {};
|
|
|
|
|
};
|
|
|
|
|
in
|
|
|
|
|
extension
|
