mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-10-03 13:30:36 +00:00
45 lines
952 B
Nix
45 lines
952 B
Nix
{
|
|
lib,
|
|
stdenvNoCC,
|
|
fetchFromGitHub,
|
|
nix-update-script,
|
|
}:
|
|
|
|
stdenvNoCC.mkDerivation {
|
|
pname = "thunderkittens";
|
|
version = "0-unstable-2026-07-13";
|
|
|
|
__structuredAttrs = true;
|
|
strictDeps = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "HazyResearch";
|
|
repo = "ThunderKittens";
|
|
rev = "1c3920d993404dd49a6d4c7267ea11d583bd5c68";
|
|
hash = "sha256-I/pzSZDx5eEimz8s+GQrQroqyQOEtNB+kas99LBM590=";
|
|
};
|
|
|
|
dontBuild = true;
|
|
|
|
installPhase = ''
|
|
runHook preInstall
|
|
|
|
mkdir -p $out/include
|
|
cp -r include/* $out/include/
|
|
|
|
runHook postInstall
|
|
'';
|
|
|
|
passthru.updateScript = nix-update-script {
|
|
extraArgs = [ "--version=branch" ];
|
|
};
|
|
|
|
meta = {
|
|
description = "Tile primitives for speedy CUDA deep learning kernels";
|
|
homepage = "https://github.com/HazyResearch/ThunderKittens";
|
|
license = lib.licenses.mit;
|
|
platforms = lib.platforms.linux;
|
|
maintainers = with lib.maintainers; [ GaetanLepage ];
|
|
};
|
|
}
|