From 273d304cf9eb866387a49a18dc0125ee713d7a8b Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 12 Mar 2025 23:07:06 +0100
Subject: [PATCH] We need gcc during runtime to enable triton to compile
 kernels.

---
 nix/docker.nix | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/nix/docker.nix b/nix/docker.nix
index c4b1d899..b622462d 100644
--- a/nix/docker.nix
+++ b/nix/docker.nix
@@ -1,4 +1,5 @@
 {
+  stdenv,
   dockerTools,
   cacert,
   text-generation-inference,
@@ -19,5 +20,8 @@ build {
     ];
 
   };
-  contents = [ cacert ];
+  contents = [
+    cacert
+    stdenv.cc
+  ];
 }