From b9ace5259442767dc8339083e315740a8d47ac3e Mon Sep 17 00:00:00 2001
From: thoffman <thoffman@embl.de>
Date: Thu, 19 Dec 2024 16:31:19 +0100
Subject: [PATCH] add tqdm dep; update jax-triton to v0.2.0

---
 ...AlphaFold3-3.0.0-foss-2024a-CUDA-12.6.0.eb |  3 +-
 ...jax-triton-0.2.0-foss-2024a-CUDA-12.6.0.eb | 36 +++++++++++++++++++
 2 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 easybuild/easyconfigs/j/jax-triton/jax-triton-0.2.0-foss-2024a-CUDA-12.6.0.eb

diff --git a/easybuild/easyconfigs/a/AlphaFold3/AlphaFold3-3.0.0-foss-2024a-CUDA-12.6.0.eb b/easybuild/easyconfigs/a/AlphaFold3/AlphaFold3-3.0.0-foss-2024a-CUDA-12.6.0.eb
index ac62633f0d2..bbf14419daf 100644
--- a/easybuild/easyconfigs/a/AlphaFold3/AlphaFold3-3.0.0-foss-2024a-CUDA-12.6.0.eb
+++ b/easybuild/easyconfigs/a/AlphaFold3/AlphaFold3-3.0.0-foss-2024a-CUDA-12.6.0.eb
@@ -28,11 +28,12 @@ builddependencies = [
 dependencies = [
     ('Python', '3.12.3'),
     ('CUDA', '12.6.0', '', SYSTEM),
+    ('tqdm', '4.67.0'),
     ('jax', '0.4.35', versionsuffix),
     ('dm-tree', '0.1.8'),
     ('dm-haiku', '0.0.13', versionsuffix),
     ('RDKit', '2024.03.5'),
-    ('jax-triton', '0.1.1', versionsuffix),
+    ('jax-triton', '0.2.0', versionsuffix),
     # TODO
 ]
 
diff --git a/easybuild/easyconfigs/j/jax-triton/jax-triton-0.2.0-foss-2024a-CUDA-12.6.0.eb b/easybuild/easyconfigs/j/jax-triton/jax-triton-0.2.0-foss-2024a-CUDA-12.6.0.eb
new file mode 100644
index 00000000000..e1dd8fe3d96
--- /dev/null
+++ b/easybuild/easyconfigs/j/jax-triton/jax-triton-0.2.0-foss-2024a-CUDA-12.6.0.eb
@@ -0,0 +1,36 @@
+# Thomas Hoffmann, EMBL Heidelberg, structures-it@embl.de, 2024/12
+
+easyblock = 'PythonBundle'
+
+name = 'jax-triton'
+version = '0.2.0'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://pypi.org/project/jax-triton'
+description = """The jax-triton repository contains integrations between JAX and Triton."""
+
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+dependencies = [
+    ('Python', '3.12.3'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('jax', '0.4.35', versionsuffix),
+    ('Triton', '3.1.0', versionsuffix),
+]
+
+use_pip = True
+
+exts_list = [
+    ('jax_triton', version, {
+        'checksums': ['ed564a5ffb9e404557dc8d296e7eb30e501da5a4d7b03408acd0837c1c618c21'],
+    }),
+]
+
+sanity_pip_check = True
+
+sanity_check_paths = {
+    'files': [],
+    'dirs': ['lib'],
+}
+
+moduleclass = 'ai'