openai-triton-llvm: init at 14.0.6-f28c006a5895

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

View File

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

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

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