From 0b706634e84558ecd75dcabbd2777b043bc3faff Mon Sep 17 00:00:00 2001
From: Antoine Viallon <antoine@lesviallon.fr>
Date: Tue, 4 Oct 2022 22:38:24 +0200
Subject: [PATCH] [Packages] Add tensorflow and rav1e support to ffmpeg 'full'

---
 overlays.nix | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/overlays.nix b/overlays.nix
index f96e371..e0fc0f6 100644
--- a/overlays.nix
+++ b/overlays.nix
@@ -128,7 +128,15 @@ in
 
         myFirefox = (import ./packages/firefox.nix { pkgs = self; inherit lib; });
 
-        kvdo = config.boot.kernelPackages.callPackage ./packages/kvdo.nix { };
+        ffmpeg-full = let
+          withLto = super.ffmpeg-full.override { enableLto = false; rav1e = self.rav1e; };
+          withTensorflow = withLto.overrideAttrs (old: {
+            CFLAGS = (old.CFLAGS or "") + " -march=${config.aviallon.general.cpuArch}";
+            LDFLAGS = (old.LDFLAGS or "") + " -march=${config.aviallon.general.cpuArch}";
+            buildInputs = (old.buildInputs or []) ++ [ super.libtensorflow-bin ];
+            configureFlags = (old.configureFlags or []) ++ [ "--enable-libtensorflow" ];
+          });
+        in withTensorflow;
       })
     ];