From 06d619eb7f1b5871a83b1892d7907bee36a71181 Mon Sep 17 00:00:00 2001
From: cfmin <1163105507@qq.com>
Date: Tue, 22 Apr 2025 16:44:41 +0800
Subject: [PATCH] Add templates app jax-0.4.30

---
 templates/jax/0.4.30/jax.arm.py39.cpu.config | 38 ++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 templates/jax/0.4.30/jax.arm.py39.cpu.config

diff --git a/templates/jax/0.4.30/jax.arm.py39.cpu.config b/templates/jax/0.4.30/jax.arm.py39.cpu.config
new file mode 100644
index 0000000..04e4a2d
--- /dev/null
+++ b/templates/jax/0.4.30/jax.arm.py39.cpu.config
@@ -0,0 +1,38 @@
+[SERVER]
+11.11.11.11
+
+[DOWNLOAD]
+jax/0.4.30 https://github.com/jax-ml/jax/archive/refs/tags/jax-v0.4.30.tar.gz
+
+[DEPENDENCY]
+module purge
+module use ./software/modulefiles
+./jarvis -install bisheng/4.1.0 com
+module load bisheng/4.1.0
+./jarvis -install python3/3.9.9 bisheng
+module load python3/3.9.9
+tar -zxvf ${JARVIS_DOWNLOAD}/jax-v0.4.30.tar.gz -C ${JARVIS_TMP_DOWNLOAD}
+
+[ENV]
+module purge
+module use ./software/modulefiles
+module load bisheng/4.1.0
+module load python3/3.9.9
+
+[APP]
+app_name = jax
+build_dir = ${JARVIS_TMP_DOWNLOAD}/jax-jax-v0.4.30
+binary_dir = 
+case_dir = ${JARVIS_TMP_DOWNLOAD}/jax-jax-v0.4.30/tests
+
+[BUILD]
+pip3 install .
+pip3 install absl-py
+
+[CLEAN]
+pip3 uninstall jax -y
+
+[RUN]
+run = python3
+binary = jax_jit_test.py
+nodes = 1
-- 
Gitee