about summary refs log tree commit diff
path: root/pkgs/development/python-modules/openai-triton/bin.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/openai-triton/bin.nix')
-rw-r--r--pkgs/development/python-modules/openai-triton/bin.nix98
1 files changed, 0 insertions, 98 deletions
diff --git a/pkgs/development/python-modules/openai-triton/bin.nix b/pkgs/development/python-modules/openai-triton/bin.nix
deleted file mode 100644
index ef95ac07244ae..0000000000000
--- a/pkgs/development/python-modules/openai-triton/bin.nix
+++ /dev/null
@@ -1,98 +0,0 @@
-{
-  lib,
-  stdenv,
-  addOpenGLRunpath,
-  cudaPackages,
-  buildPythonPackage,
-  fetchurl,
-  isPy38,
-  isPy39,
-  isPy310,
-  isPy311,
-  python,
-  autoPatchelfHook,
-  filelock,
-  lit,
-  pythonRelaxDepsHook,
-  zlib,
-}:
-
-buildPythonPackage rec {
-  pname = "triton";
-  version = "2.1.0";
-  format = "wheel";
-
-  src =
-    let
-      pyVerNoDot = lib.replaceStrings [ "." ] [ "" ] python.pythonVersion;
-      unsupported = throw "Unsupported system";
-      srcs = (import ./binary-hashes.nix version)."${stdenv.system}-${pyVerNoDot}" or unsupported;
-    in
-    fetchurl srcs;
-
-  disabled = !(isPy38 || isPy39 || isPy310 || isPy311);
-
-  pythonRemoveDeps = [
-    "cmake"
-    "torch"
-  ];
-
-  buildInputs = [ zlib ];
-
-  nativeBuildInputs = [
-    pythonRelaxDepsHook # torch and triton refer to each other so this hook is included to mitigate that.
-    autoPatchelfHook
-  ];
-
-  propagatedBuildInputs = [
-    filelock
-    lit
-    zlib
-  ];
-
-  dontStrip = true;
-
-  # If this breaks, consider replacing with "${cuda_nvcc}/bin/ptxas"
-  postFixup =
-    ''
-      chmod +x "$out/${python.sitePackages}/triton/third_party/cuda/bin/ptxas"
-    ''
-    + (
-      let
-        # Bash was getting weird without linting,
-        # but basically upstream contains [cc, ..., "-lcuda", ...]
-        # and we replace it with [..., "-lcuda", "-L/run/opengl-driver/lib", "-L$stubs", ...]
-        old = [ "-lcuda" ];
-        new = [
-          "-lcuda"
-          "-L${addOpenGLRunpath.driverLink}"
-          "-L${cudaPackages.cuda_cudart}/lib/stubs/"
-        ];
-
-        quote = x: ''"${x}"'';
-        oldStr = lib.concatMapStringsSep ", " quote old;
-        newStr = lib.concatMapStringsSep ", " quote new;
-      in
-      ''
-        substituteInPlace $out/${python.sitePackages}/triton/common/build.py \
-          --replace '${oldStr}' '${newStr}'
-      ''
-    );
-
-  meta = with lib; {
-    description = "Language and compiler for custom Deep Learning operations";
-    homepage = "https://github.com/openai/triton/";
-    changelog = "https://github.com/openai/triton/releases/tag/v${version}";
-    # Includes NVIDIA's ptxas, but redistributions of the binary are not limited.
-    # https://docs.nvidia.com/cuda/eula/index.html
-    # triton's license is MIT.
-    # openai-triton-bin includes ptxas binary, therefore unfreeRedistributable is set.
-    license = with licenses; [
-      unfreeRedistributable
-      mit
-    ];
-    sourceProvenance = with sourceTypes; [ binaryNativeCode ];
-    platforms = [ "x86_64-linux" ];
-    maintainers = with maintainers; [ junjihashimoto ];
-  };
-}