From 3be497344abe5e90269d10a0acabf280896cd6e8 Mon Sep 17 00:00:00 2001 From: Auriane R <48684432+aurianer@users.noreply.github.com> Date: Fri, 21 Jun 2024 17:57:58 +0200 Subject: [PATCH] py-triton: add main (#44774) * Add triton env variable to build with clang * Add git url and main branch for py-triton --- var/spack/repos/builtin/packages/py-triton/package.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/var/spack/repos/builtin/packages/py-triton/package.py b/var/spack/repos/builtin/packages/py-triton/package.py index 2740063ef69..aecefdf9b87 100644 --- a/var/spack/repos/builtin/packages/py-triton/package.py +++ b/var/spack/repos/builtin/packages/py-triton/package.py @@ -11,14 +11,22 @@ class PyTriton(PythonPackage): homepage = "https://github.com/openai/triton" url = "https://github.com/openai/triton/archive/refs/tags/v2.1.0.tar.gz" + git = "https://github.com/openai/triton.git" license("MIT") + version("main", branch="main") version("2.1.0", sha256="4338ca0e80a059aec2671f02bfc9320119b051f378449cf5f56a1273597a3d99") depends_on("py-setuptools@40.8:", type="build") depends_on("cmake@3.18:", type="build") depends_on("py-filelock", type=("build", "run")) depends_on("zlib-api", type="link") + conflicts("^openssl@3.3.0") + + def setup_build_environment(self, env): + """Set environment variables used to control the build""" + if self.spec.satisfies("%clang"): + env.set("TRITON_BUILD_WITH_CLANG_LLD", "True") build_directory = "python"