diff --git a/templates/top50/jax/0.4.30/jax.arm.py39.cpu.config b/templates/top50/jax/0.4.30/jax.arm.py39.cpu.config new file mode 100644 index 0000000000000000000000000000000000000000..a5a098542755c9e92352a185491aec123dcc94fb --- /dev/null +++ b/templates/top50/jax/0.4.30/jax.arm.py39.cpu.config @@ -0,0 +1,39 @@ +[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 +