{ stdenv, llvmPackages, lib, fetchFromGitHub, cmake, flatbuffers, libffi, libpng, libjpeg, mesa, libGL, eigen, openblas, blas, lapack, pythonSupport ? false, python3Packages ? null, }: assert blas.implementation == "openblas" && lapack.implementation == "openblas"; stdenv.mkDerivation rec { pname = "halide"; version = "18.0.0"; src = fetchFromGitHub { owner = "halide"; repo = "Halide"; rev = "v${version}"; hash = "sha256-BPalUh9EgdCqVaWC1HoreyyRcPQc4QMIYnLrRoNDDCI="; }; postPatch = '' # See https://github.com/halide/Halide/issues/7785 substituteInPlace 'src/runtime/HalideRuntime.h' \ --replace '#if defined(__x86_64__) || defined(__i386__) || defined(__arm__) || defined(__aarch64__) #define HALIDE_CPP_COMPILER_HAS_FLOAT16' \ '#if defined(__x86_64__) || defined(__i386__) #define HALIDE_CPP_COMPILER_HAS_FLOAT16' '' # Note: on x86_64-darwin, clang fails to find AvailabilityVersions.h, so we remove it. # Halide uses AvailabilityVersions.h and TargetConditionals.h to determine whether # ::aligned_alloc is available. For us, it isn't. + lib.optionalString (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) '' substituteInPlace 'src/runtime/HalideBuffer.h' \ --replace '#ifdef __APPLE__ #include #include #endif' \ ' ' \ --replace 'TARGET_OS_OSX && (__MAC_OS_X_VERSION_MIN_REQUIRED < __MAC_10_15)' \ '1' \ --replace 'TARGET_OS_IPHONE && (__IPHONE_OS_VERSION_MIN_REQUIRED < __IPHONE_14_0)' \ '0' ''; cmakeFlags = [ "-DWARNINGS_AS_ERRORS=OFF" "-DWITH_PYTHON_BINDINGS=${if pythonSupport then "ON" else "OFF"}" "-DTARGET_WEBASSEMBLY=OFF" # Disable performance tests since they may fail on busy machines "-DWITH_TEST_PERFORMANCE=OFF" # Disable fuzzing tests -- this has become the default upstream after the # v16 release (See https://github.com/halide/Halide/commit/09c5d1d19ec8e6280ccbc01a8a12decfb27226ba) # These tests also fail to compile on Darwin because of some missing command line options... "-DWITH_TEST_FUZZ=OFF" # Disable FetchContent for flatbuffers and use the version from nixpkgs instead "-DFLATBUFFERS_USE_FETCHCONTENT=OFF" "-DPYBIND11_USE_FETCHCONTENT=OFF" ]; doCheck = true; preCheck = let disabledTests = lib.strings.concatStringsSep "|" [ # Requires too much parallelism for remote builders. "mullapudi2016_fibonacci" # Take too long---we don't want to run these in CI. "adams2019_test_apps_autoscheduler" "anderson2021_test_apps_autoscheduler" "correctness_cross_compilation" "correctness_simd_op_check_hvx" ]; in '' checkFlagsArray+=("ARGS=-E '${disabledTests}'") ''; postInstall = lib.optionalString pythonSupport '' mkdir -p $out/${builtins.dirOf python3Packages.python.sitePackages} mv -v $out/lib/python3/site-packages $out/${python3Packages.python.sitePackages} rmdir $out/lib/python3/ ''; # Note: only openblas and not atlas part of this Nix expression # see pkgs/development/libraries/science/math/liblapack/3.5.0.nix # to get a hint howto setup atlas instead of openblas buildInputs = [ llvmPackages.llvm llvmPackages.lld llvmPackages.openmp llvmPackages.libclang libffi libpng libjpeg eigen openblas ] ++ lib.optionals (!stdenv.hostPlatform.isDarwin) [ mesa libGL ]; nativeBuildInputs = [ cmake flatbuffers ] ++ lib.optionals pythonSupport [ python3Packages.python python3Packages.pybind11 ]; propagatedBuildInputs = lib.optionals pythonSupport [ python3Packages.numpy python3Packages.imageio ]; meta = with lib; { description = "C++ based language for image processing and computational photography"; homepage = "https://halide-lang.org"; license = licenses.mit; platforms = platforms.all; maintainers = with maintainers; [ ck3d atila twesterhout ]; }; }