Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 65 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566{ lib, bootstrapData, db,}:
bootstrapData// { /** All CUDA capabilities, sorted by version.
NOTE: Since the capabilities are sorted by version and architecture/family-specific features are appended to the minor version component, the sorted list groups capabilities by baseline feature set.
# Type
``` allSortedCudaCapabilities :: [CudaCapability] ```
# Example
``` allSortedCudaCapabilities = [ "5.0" "5.2" "6.0" "6.1" "7.0" "7.2" "7.5" "8.0" "8.6" "8.7" "8.9" "9.0" "9.0a" "10.0" "10.0a" "10.0f" "10.1" "10.1a" "10.1f" "10.3" "10.3a" "10.3f" ]; ``` */ allSortedCudaCapabilities = lib.sort lib.versionOlder (lib.attrNames db.cudaCapabilityToInfo);
/** Mapping of CUDA micro-architecture name to capabilities belonging to that micro-architecture.
# Type
``` cudaArchNameToCapabilities :: AttrSet NonEmptyStr (NonEmptyListOf CudaCapability) ``` */ cudaArchNameToCapabilities = lib.groupBy ( cudaCapability: db.cudaCapabilityToInfo.${cudaCapability}.archName ) db.allSortedCudaCapabilities;}