nixpkgs/pkgs/development/haskell-modules/configuration-tensorflow.nix
Mike Sperber 1df0e06d54 haskell-tensorflow: unbreak build
This just moves the package to the current head in the official repo,
which has since seen patches that unbreak the ghc-9 build and work
with our version of TensorFlow.
2023-02-23 08:55:42 +01:00

38 lines
1.3 KiB
Nix

{ pkgs, haskellLib }:
with haskellLib;
self: super:
let
# This contains updates to the dependencies, without which it would
# be even more work to get it to build.
# As of 2020-04, there's no new release in sight, which is why we're
# pulling from Github.
tensorflow-haskell = pkgs.fetchFromGitHub {
owner = "tensorflow";
repo = "haskell";
rev = "555d90c43202d5a3021893013bfc8e2ffff58c97";
sha256 = "uOuIeD4o+pcjvluTqyVU3GJUQ4e1+p3FhINJ9b6oK+k=";
fetchSubmodules = true;
};
setTensorflowSourceRoot = dir: drv:
(overrideCabal (drv: { src = tensorflow-haskell; }) drv)
.overrideAttrs (_oldAttrs: {sourceRoot = "source/${dir}";});
in
{
tensorflow-proto = doJailbreak (setTensorflowSourceRoot "tensorflow-proto" super.tensorflow-proto);
tensorflow = overrideCabal
(drv: { libraryHaskellDepends = drv.libraryHaskellDepends ++ [self.vector-split]; })
(setTensorflowSourceRoot "tensorflow" super.tensorflow);
tensorflow-core-ops = setTensorflowSourceRoot "tensorflow-core-ops" super.tensorflow-core-ops;
tensorflow-logging = setTensorflowSourceRoot "tensorflow-logging" super.tensorflow-logging;
tensorflow-opgen = setTensorflowSourceRoot "tensorflow-opgen" super.tensorflow-opgen;
tensorflow-ops = setTensorflowSourceRoot "tensorflow-ops" super.tensorflow-ops;
}