zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-triton: add main (#44774)

Browse Source 
* Add triton env variable to build with clang

* Add git url and main branch for py-triton
This commit is contained in:
Auriane R 2024-06-21 17:57:58 +02:00 committed by GitHub
parent 05413689b9
commit 3be497344a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/py-triton/package.py
View File

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