summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-triton/package.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-triton/package.py b/var/spack/repos/builtin/packages/py-triton/package.py
index 2740063ef6..aecefdf9b8 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"