From 1fae17d6fbc40bbc0e3cfe19fb825857edebf2b9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AB=98=E6=96=87=E5=B0=A7?= <18848977154@163.com>
Date: Wed, 17 Dec 2025 20:08:47 +0800
Subject: [PATCH] jax

---
 templates/jax/0.4.30/jax.arm.py39.cpu.config | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/templates/jax/0.4.30/jax.arm.py39.cpu.config b/templates/jax/0.4.30/jax.arm.py39.cpu.config
index 0a6a314a..2361783f 100644
--- a/templates/jax/0.4.30/jax.arm.py39.cpu.config
+++ b/templates/jax/0.4.30/jax.arm.py39.cpu.config
@@ -9,16 +9,19 @@ module purge
 ./jarvis -install hpckit/${HPCKIT_VERSION} any
 module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles
 module load bisheng/compiler${BISHENG_VERSION}/bishengmodule
-
+module load bisheng/hmpi${HPCKIT_VERSION}/release
 ./jarvis -install python3/3.9.9 bisheng
+module use ./software/moduledeps/bisheng${BISHENG_VERSION}
 module load python3/3.9.9
+rm -rf ${JARVIS_TMP_DOWNLOAD}/jax-v0.4.30
 tar -zxvf ${JARVIS_DOWNLOAD}/jax-v0.4.30.tar.gz -C ${JARVIS_TMP_DOWNLOAD}
 
 [ENV]
 module purge
 module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles
 module load bisheng/compiler${BISHENG_VERSION}/bishengmodule
-
+module load bisheng/hmpi${HPCKIT_VERSION}/release
+module use ./software/moduledeps/bisheng${BISHENG_VERSION}
 module load python3/3.9.9
 
 [APP]
@@ -38,3 +41,4 @@ pip3 uninstall jax -y
 run = python3
 binary = jax_jit_test.py
 nodes = 1
+
-- 
Gitee