diff options
Diffstat (limited to 'pkgs/development/python-modules/openai-triton/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/openai-triton/default.nix | 3 |
1 files changed, 1 insertions, 2 deletions
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"; |