mirrors/nixpkgs
1
0
Fork 1
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-18 11:40:45 +00:00
Code Issues Wiki Activity

openai-triton-llvm: init at 14.0.6-f28c006a5895

Browse source 
Moved from python3Packages.openai-triton due to rebuilds per python version
This commit is contained in:
Madoura 2023-10-23 18:39:33 -05:00
parent ac0b05fe7f
commit b5dce72180
No known key found for this signature in database
GPG key ID: 1B9BB76A2B71922D
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pkgs/development/python-modules/openai-triton/llvm.nix → pkgs/by-name/op/openai-triton-llvm/package.nix
View file

@ -22,7 +22,7 @@
}:
stdenv.mkDerivation (finalAttrs: {
pname = "triton-llvm";
pname = "openai-triton-llvm";
version = "14.0.6-f28c006a5895";
outputs = [

3
pkgs/development/python-modules/openai-triton/default.nix
View file

@ -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";

1
pkgs/top-level/python-packages.nix
View file

@ -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;
};