已合并
jax #924
gao-wenyao123创建于 2025年12月17日
jax #924
已合并
从refs/pull/924/head合入到master
共 1 个文件变更+6-2
| @@ -9,16 +9,19 @@ module purge | |||
| 9 | ./jarvis -install hpckit/${HPCKIT_VERSION} any | 9 | ./jarvis -install hpckit/${HPCKIT_VERSION} any |
| 10 | module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles | 10 | module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles |
| 11 | module load bisheng/compiler${BISHENG_VERSION}/bishengmodule | 11 | module load bisheng/compiler${BISHENG_VERSION}/bishengmodule |
| 12 | - | 12 | +module load bisheng/hmpi${HPCKIT_VERSION}/release |
| 13 | ./jarvis -install python3/3.9.9 bisheng | 13 | ./jarvis -install python3/3.9.9 bisheng |
| 14 | +module use ./software/moduledeps/bisheng${BISHENG_VERSION} | ||
| 14 | module load python3/3.9.9 | 15 | module load python3/3.9.9 |
| 16 | +rm -rf ${JARVIS_TMP_DOWNLOAD}/jax-v0.4.30 | ||
| 15 | tar -zxvf ${JARVIS_DOWNLOAD}/jax-v0.4.30.tar.gz -C ${JARVIS_TMP_DOWNLOAD} | 17 | tar -zxvf ${JARVIS_DOWNLOAD}/jax-v0.4.30.tar.gz -C ${JARVIS_TMP_DOWNLOAD} |
| 16 | 18 | ||
| 17 | [ENV] | 19 | [ENV] |
| 18 | module purge | 20 | module purge |
| 19 | module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles | 21 | module use ./software/utils/hpckit/${HPCKIT_VERSION}/HPCKit/${HPCKIT_VERSION}/modulefiles |
| 20 | module load bisheng/compiler${BISHENG_VERSION}/bishengmodule | 22 | module load bisheng/compiler${BISHENG_VERSION}/bishengmodule |
| 21 | - | 23 | +module load bisheng/hmpi${HPCKIT_VERSION}/release |
| 24 | +module use ./software/moduledeps/bisheng${BISHENG_VERSION} | ||
| 22 | module load python3/3.9.9 | 25 | module load python3/3.9.9 |
| 23 | 26 | ||
| 24 | [APP] | 27 | [APP] |
| @@ -38,3 +41,4 @@ pip3 uninstall jax -y | |||
| 38 | run = python3 | 41 | run = python3 |
| 39 | binary = jax_jit_test.py | 42 | binary = jax_jit_test.py |
| 40 | nodes = 1 | 43 | nodes = 1 |
| 44 | + | ||