Gaetan Lepage
2025-07-08 15:09:34 +02:00
parent 6106c71096
commit f7fba33006
2 changed files with 39 additions and 27 deletions
@@ -15,7 +15,7 @@
buildPythonPackage rec {
pname = "triton";
version = "3.1.0";
version = "3.3.1";
format = "wheel";
src =
@@ -26,8 +26,6 @@ buildPythonPackage rec {
in
fetchurl srcs;
disabled = pythonOlder "3.8";
pythonRemoveDeps = [
"cmake"
# torch and triton refer to each other so this hook is included to mitigate that.
@@ -54,7 +52,7 @@ buildPythonPackage rec {
ln -s ${cudaPackages.cuda_nvcc}/bin/ptxas $out/${python.sitePackages}/triton/third_party/cuda/bin/
'';
meta = with lib; {
meta = {
description = "Language and compiler for custom Deep Learning operations";
homepage = "https://github.com/triton-lang/triton/";
changelog = "https://github.com/triton-lang/triton/releases/tag/v${version}";
@@ -62,12 +60,11 @@ buildPythonPackage rec {
# https://docs.nvidia.com/cuda/eula/index.html
# triton's license is MIT.
# triton-bin includes ptxas binary, therefore unfreeRedistributable is set.
license = with licenses; [
license = with lib.licenses; [
unfreeRedistributable
mit
];
sourceProvenance = with sourceTypes; [ binaryNativeCode ];
platforms = [ "x86_64-linux" ];
maintainers = with maintainers; [ junjihashimoto ];
sourceProvenance = with lib.sourceTypes; [ binaryNativeCode ];
maintainers = with lib.maintainers; [ junjihashimoto ];
};
}
@@ -7,31 +7,46 @@
version:
builtins.getAttr version {
"3.1.0" = {
x86_64-linux-38 = {
name = "triton-3.1.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.1.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
hash = "sha256-ba2sp/wk3jThgCcbXPhkwWdVcC6fY6FvYt9xSoCZEmo=";
};
x86_64-linux-39 = {
name = "triton-3.1.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl ";
url = "https://download.pytorch.org/whl/triton-3.1.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
hash = "sha256-qvqaIM0Nn+5SPNRQSqcTGAeoZM133Pbv5+mB8YuMbBE=";
"3.3.1" = {
aarch64-linux-310 = {
name = "triton-3.3.1-cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
hash = "sha256-nuTCrhQM7ftvnmnlRFf6KIqrbxOP0axLcdqSzfpmCQE=";
};
x86_64-linux-310 = {
name = "triton-3.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl ";
url = "https://download.pytorch.org/whl/triton-3.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
hash = "sha256-aw3RCpJSY6u+n6N9zeZ6Xpsjg/wmn99Z9WV8rDjF0dg=";
name = "triton-3.3.1-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
hash = "sha256-t020RbHFYoRNPPrW6Weccuk/37GpCiQFKwO7XEnRJC4=";
};
aarch64-linux-311 = {
name = "triton-3.3.1-cp311-cp311-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp311-cp311-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
hash = "sha256-1OMWwp3LUQPc/N+Y3o+huRwL+yYhisV37MwwWFrftg0=";
};
x86_64-linux-311 = {
name = "triton-3.1.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.1.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
hash = "sha256-DzT254hdG/Dqr3uodaXwzm88E7qY+VA2UcHm3GdX7Vw=";
name = "triton-3.3.1-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
hash = "sha256-sx46om+Ms8xb9OGHv3N8us8XMR4REreB1KBZNT39cxs=";
};
aarch64-linux-312 = {
name = "triton-3.3.1-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
hash = "sha256-mxmTw7pHARfc8oiP8RbYKAntek080FB0/ED/ORkLLOg=";
};
x86_64-linux-312 = {
name = "triton-3.1.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.1.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl";
hash = "sha256-yBgvQv2AgKfTnWZoFPo2xeMMwA6n7usaKYPbtMmaD9w=";
name = "triton-3.3.1-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
hash = "sha256-mZnoOroh4aeMHzbyG85iG3e8qlMCd6UEhKfLSoIvbkM=";
};
aarch64-linux-313 = {
name = "triton-3.3.1-cp313-cp313-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp313-cp313-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl";
hash = "sha256-S3nkmxWuMd0JWQkPN6Ni64HHfFZDug7FblKZA7BtKM8=";
};
x86_64-linux-313 = {
name = "triton-3.3.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
url = "https://download.pytorch.org/whl/triton-3.3.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl";
hash = "sha256-uJ2Ea1pBmDF/7Cel06YJ6pa21Vf/RLVsIxdlRgI8QkA=";
};
};
}