94 lines
2.6 KiB
Nix
94 lines
2.6 KiB
Nix
{
|
|
# Build Pyo3 package
|
|
inputs = {
|
|
nixpkgs.url = "github:nixos/nixpkgs/nixpkgs-unstable";
|
|
flake-utils.url = "github:numtide/flake-utils";
|
|
rust-overlay = {
|
|
url = "github:oxalica/rust-overlay";
|
|
inputs.nixpkgs.follows = "nixpkgs";
|
|
};
|
|
crane.url = "github:ipetkov/crane";
|
|
fenix = {
|
|
url = "github:nix-community/fenix";
|
|
inputs.nixpkgs.follows = "nixpkgs";
|
|
};
|
|
};
|
|
|
|
outputs =
|
|
inputs:
|
|
inputs.flake-utils.lib.eachDefaultSystem (
|
|
system:
|
|
let
|
|
pkgs = import inputs.nixpkgs {
|
|
inherit system;
|
|
overlays = [ inputs.rust-overlay.overlays.default ];
|
|
config = {
|
|
allowUnfree = true;
|
|
cudaSupport = false;
|
|
};
|
|
};
|
|
lib = pkgs.lib;
|
|
|
|
python_version = pkgs.python313;
|
|
wheel_tail = "cp313-cp313-linux_x86_64"; # Change if python_version changes
|
|
|
|
# Get a custom rust toolchain
|
|
craneLib = (inputs.crane.mkLib pkgs).overrideToolchain (
|
|
p: inputs.fenix.packages.${system}.complete.toolchain
|
|
);
|
|
|
|
project_name = (craneLib.crateNameFromCargoToml { cargoToml = ./shuffles/Cargo.toml; }).pname;
|
|
project_version = (craneLib.crateNameFromCargoToml { cargoToml = ./shuffles/Cargo.toml; }).version;
|
|
|
|
crate_cfg = {
|
|
src =
|
|
let
|
|
fs = lib.fileset;
|
|
in
|
|
fs.toSource {
|
|
root = ./shuffles;
|
|
fileset = fs.unions [
|
|
./shuffles/src/lib.rs
|
|
./shuffles/Cargo.lock
|
|
./shuffles/Cargo.toml
|
|
./shuffles/pyproject.toml
|
|
];
|
|
};
|
|
nativeBuildInputs = [ python_version ];
|
|
# doCheck = true;
|
|
# buildInputs = [];
|
|
};
|
|
|
|
crate_artifacts = craneLib.buildDepsOnly (
|
|
crate_cfg
|
|
// {
|
|
pname = "${project_name}-artifacts";
|
|
version = project_version;
|
|
}
|
|
);
|
|
|
|
# Build the library, then re-use the target dir to generate the wheel file with maturin
|
|
crate_pkg = (
|
|
craneLib.buildPackage (
|
|
crate_cfg
|
|
// {
|
|
pname = project_name;
|
|
version = project_version;
|
|
cargoArtifacts = crate_artifacts;
|
|
}
|
|
)
|
|
);
|
|
in
|
|
rec {
|
|
devShells.default = craneLib.devShell {
|
|
packages = [
|
|
pkgs.cargo
|
|
pkgs.cargo-flamegraph
|
|
];
|
|
shellHook = ''
|
|
export CUDA_PATH=${pkgs.cudatoolkit}
|
|
'';
|
|
};
|
|
}
|
|
);
|
|
}
|