mirror of
https://github.com/huggingface/kernels.git
synced 2025-10-20 12:33:46 +08:00
b0c431fee4
* Add the `kernels check` subcommand This subcommand checks a given kernel. Currently it applies the same ABI checks as `kernel-abi-check` in `kernel-builder`. * Print an error when `build` contains files * Forgot to update has_issues in two places
65 lines
1.5 KiB
Nix
65 lines
1.5 KiB
Nix
{
|
|
inputs = {
|
|
hf-nix.url = "github:huggingface/hf-nix";
|
|
nixpkgs.follows = "hf-nix/nixpkgs";
|
|
flake-utils.url = "github:numtide/flake-utils";
|
|
};
|
|
outputs =
|
|
{
|
|
self,
|
|
nixpkgs,
|
|
flake-utils,
|
|
hf-nix,
|
|
}:
|
|
flake-utils.lib.eachDefaultSystem (
|
|
system:
|
|
let
|
|
pkgs = import nixpkgs {
|
|
inherit system;
|
|
config = hf-nix.lib.config system;
|
|
overlays = [
|
|
hf-nix.overlays.default
|
|
];
|
|
};
|
|
in
|
|
{
|
|
formatter = pkgs.nixfmt-tree;
|
|
packages.kernel-abi-check = pkgs.python3.pkgs.callPackage ./nix/kernel-abi-check.nix {};
|
|
devShells = with pkgs; rec {
|
|
default = mkShell {
|
|
nativeBuildInputs = [
|
|
# For hf-doc-builder.
|
|
nodejs
|
|
];
|
|
buildInputs =
|
|
[
|
|
black
|
|
mypy
|
|
pyright
|
|
ruff
|
|
]
|
|
++ (with python3.pkgs; [
|
|
docutils
|
|
huggingface-hub
|
|
(callPackage ./nix/kernel-abi-check.nix {})
|
|
mktestdocs
|
|
pytest
|
|
pytest-benchmark
|
|
pyyaml
|
|
torch
|
|
types-pyyaml
|
|
venvShellHook
|
|
]);
|
|
|
|
venvDir = "./.venv";
|
|
|
|
postVenvCreation = ''
|
|
unset SOURCE_DATE_EPOCH
|
|
( python -m pip install --no-build-isolation --no-dependencies -e . )
|
|
'';
|
|
};
|
|
};
|
|
}
|
|
);
|
|
}
|