diff options
-rw-r--r-- | pkgs/by-name/op/openai-triton-llvm/package.nix (renamed from pkgs/development/python-modules/openai-triton/llvm.nix) | 2 | ||||
-rw-r--r-- | pkgs/development/python-modules/openai-triton/default.nix | 3 | ||||
-rw-r--r-- | pkgs/top-level/python-packages.nix | 1 |
3 files changed, 3 insertions, 3 deletions
diff --git a/pkgs/development/python-modules/openai-triton/llvm.nix b/pkgs/by-name/op/openai-triton-llvm/package.nix index 70ea69a9b15..2fb56d0a635 100644 --- a/pkgs/development/python-modules/openai-triton/llvm.nix +++ b/pkgs/by-name/op/openai-triton-llvm/package.nix @@ -22,7 +22,7 @@ }: stdenv.mkDerivation (finalAttrs: { - pname = "triton-llvm"; + pname = "openai-triton-llvm"; version = "14.0.6-f28c006a5895"; outputs = [ diff --git a/pkgs/development/python-modules/openai-triton/default.nix b/pkgs/development/python-modules/openai-triton/default.nix index 89545f03265..e1ac9cb4cef 100644 --- a/pkgs/development/python-modules/openai-triton/default.nix +++ b/pkgs/development/python-modules/openai-triton/default.nix @@ -1,5 +1,4 @@ { lib -, callPackage , config , buildPythonPackage , fetchFromGitHub @@ -15,6 +14,7 @@ , ncurses , libxml2 , lit +, llvm , filelock , torchWithRocm , python @@ -35,7 +35,6 @@ let # be executed on the GPU. # Cf. https://nixos.org/manual/nixpkgs/unstable/#sec-cross-infra ptxas = "${pkgsTargetTarget.cudaPackages.cuda_nvcc}/bin/ptxas"; # Make sure cudaPackages is the right version each update (See python/setup.py) - llvm = callPackage ./llvm.nix { }; # Use a custom llvm, see llvm.nix for details in buildPythonPackage rec { pname = "triton"; diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 31e66b5fa91..9d4be43e17d 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -8430,6 +8430,7 @@ self: super: with self; { open-meteo = callPackage ../development/python-modules/open-meteo { }; openai-triton = callPackage ../development/python-modules/openai-triton { + llvm = pkgs.openai-triton-llvm; cudaPackages = pkgs.cudaPackages_12_0; }; |