mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-04 12:53:05 +00:00
90 lines
2.5 KiB
Nix
90 lines
2.5 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
cmake,
|
|
ninja,
|
|
fetchFromGitHub,
|
|
fetchpatch,
|
|
python3,
|
|
opencv,
|
|
nlohmann_json,
|
|
nanoflann,
|
|
glm,
|
|
cxxopts,
|
|
config,
|
|
# Upstream has rocm/hip support, too. anyone?
|
|
cudaSupport ? config.cudaSupport,
|
|
cudaPackages,
|
|
autoAddDriverRunpath,
|
|
}:
|
|
let
|
|
version = "1.1.2";
|
|
torch = python3.pkgs.torch.override { inherit cudaSupport; };
|
|
# Using a normal stdenv with cuda torch gives
|
|
# ld: /nix/store/k1l7y96gv0nc685cg7i3g43i4icmddzk-python3.11-torch-2.2.1-lib/lib/libc10.so: undefined reference to `std::ios_base_library_init()@GLIBCXX_3.4.32'
|
|
stdenv' = if cudaSupport then cudaPackages.backendStdenv else stdenv;
|
|
in
|
|
stdenv'.mkDerivation {
|
|
pname = "opensplat";
|
|
inherit version;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "pierotofy";
|
|
repo = "OpenSplat";
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-3tk62b5fSf6wzuc5TwkdfAKgUMrw3ZxetCJa2RVMS/s=";
|
|
};
|
|
|
|
patches = [
|
|
(fetchpatch {
|
|
name = "install-executables.patch";
|
|
url = "https://github.com/pierotofy/OpenSplat/commit/b4c4176819b508978583b7ebf66306171807a8e6.patch";
|
|
hash = "sha256-BUgPMcO3lt3ZEzv24u36k3aTEIoloOhxrCGi1KQ5Epk=";
|
|
})
|
|
];
|
|
|
|
postPatch = ''
|
|
# the two vendored gsplats are so heavily modified they may be considered a fork
|
|
find vendor ! -name 'gsplat*' -maxdepth 1 -mindepth 1 -exec rm -rf {} +
|
|
mkdir vendor/{nanoflann,glm}
|
|
ln -s ${glm}/include/glm vendor/glm/glm
|
|
ln -s ${nanoflann}/include/nanoflann.hpp vendor/nanoflann/nanoflann.hpp
|
|
ln -s ${nlohmann_json}/include/nlohmann vendor/json
|
|
ln -s ${cxxopts}/include/cxxopts.hpp vendor/cxxopts.hpp
|
|
'';
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
ninja
|
|
] ++ lib.optionals cudaSupport [
|
|
cudaPackages.cuda_nvcc
|
|
autoAddDriverRunpath
|
|
];
|
|
|
|
buildInputs = [
|
|
nlohmann_json
|
|
torch.cxxdev
|
|
torch
|
|
opencv
|
|
] ++ lib.optionals cudaSupport [
|
|
cudaPackages.cuda_cudart
|
|
];
|
|
|
|
env.TORCH_CUDA_ARCH_LIST = "${lib.concatStringsSep ";" python3.pkgs.torch.cudaCapabilities}";
|
|
|
|
cmakeFlags = [
|
|
(lib.cmakeBool "CMAKE_SKIP_RPATH" true)
|
|
] ++ lib.optionals cudaSupport [
|
|
(lib.cmakeFeature "GPU_RUNTIME" "CUDA")
|
|
(lib.cmakeFeature "CUDA_TOOLKIT_ROOT_DIR" "${cudaPackages.cudatoolkit}/")
|
|
];
|
|
|
|
meta = {
|
|
description = "Production-grade 3D gaussian splatting";
|
|
homepage = "https://github.com/pierotofy/OpenSplat/";
|
|
license = lib.licenses.mit;
|
|
maintainers = [ lib.maintainers.jcaesar ];
|
|
platforms = lib.platforms.linux ++ lib.optionals (!cudaSupport) lib.platforms.darwin;
|
|
};
|
|
}
|