Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

rebase jax and update machine #31

Merged
merged 7 commits into from
May 6, 2024
Merged

rebase jax and update machine #31

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
dac004d
rebase jax and update machine
wendyliu235 Apr 25, 2024
c26608f
change to 0.4.24
wendyliu235 Apr 25, 2024
0ee64bc
Merge branch 'main' into machine_update
wendyliu235 Apr 28, 2024
1cdb62c
change requirement install and ut branch
wendyliu235 Apr 28, 2024
df76f26
Update IDC_1100_Public_CI.yml
wendyliu235 Apr 28, 2024
5b2f43b
Update IDC_1100_Public_CI.yml
wendyliu235 Apr 29, 2024
d958649
Update IDC_1100_Public_CI.yml
wendyliu235 May 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 11 additions & 9 deletions .github/workflows/IDC_1100_Public_CI.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,14 @@ jobs:
run: |
set +e
CONDA_ENV=openxla_preCI_${{ github.event.pull_request.number }}_${{ runner.name }}
source /opt/miniconda/bin/activate ${CONDA_ENV}
source /opt/conda/bin/activate ${CONDA_ENV}
if [ $? -ne 0 ]; then
echo 'conda env does not exist'
conda create -n ${CONDA_ENV} python=3.10 -y
conda activate ${CONDA_ENV}
fi
pip install jax==0.4.20 jaxlib==0.4.20
pip install -r ./test/requirements.txt
#pip install jax==0.4.24 jaxlib==0.4.24
conda install libstdcxx-ng==12.2.0 -c conda-forge
pip install absl-py
pip list | grep numpy
Expand All @@ -40,7 +41,7 @@ jobs:
run: |
set +e
CONDA_ENV=openxla_preCI_${{ github.event.pull_request.number }}_${{ runner.name }}
source /opt/miniconda/bin/activate ${CONDA_ENV}
source /opt/conda/bin/activate ${CONDA_ENV}
which bazel
if [ $? -eq 1 ]; then
wget https://github.com/bazelbuild/bazel/releases/download/5.3.0/bazel-5.3.0-installer-linux-x86_64.sh
Expand Down Expand Up @@ -70,7 +71,7 @@ jobs:
echo $build_result
if [ "$build_result" = "0" ];then
echo "Build successful"
pip install *.whl
pip install --force-reinstall *.whl
else
echo "Build Failed"
exit 1
Expand All @@ -79,13 +80,14 @@ jobs:
run: |
Zantares marked this conversation as resolved.
Show resolved Hide resolved
set +e
CONDA_ENV=openxla_preCI_${{ github.event.pull_request.number }}_${{ runner.name }}
source /opt/miniconda/bin/activate ${CONDA_ENV}
source /opt/conda/bin/activate ${CONDA_ENV}
basekit_path=/home/sdp
source ${basekit_path}/intel/oneapi/compiler/latest/env/vars.sh
source ${basekit_path}/intel/oneapi/mkl/latest/env/vars.sh
export ZE_AFFINITY_MASK=0,1
export ZE_AFFINITY_MASK=5,6
ut_branch=$(cat ./test/BRANCH_NAME)
git clone https://github.com/google/jax.git
cd jax && git checkout jaxlib-v0.4.20
cd jax && git checkout $ut_branch
patch -p1 -i ../test/jax_ut.patch
pip install hypothesis
pip install -e .
Expand Down Expand Up @@ -170,10 +172,10 @@ jobs:
uses: actions/upload-artifact@v3
with:
name: XLA-CI-Data
path: /home/sdp/actions-runner/workspace_openxla/intel-extension-for-openxla/intel-extension-for-openxla/jax/logs
path: /home/sdp/actions-runner/_work/intel-extension-for-openxla/intel-extension-for-openxla/jax/logs
- name: Test Results Check
run: |
cd /home/sdp/actions-runner/workspace_openxla/intel-extension-for-openxla/intel-extension-for-openxla/jax
cd /home/sdp/actions-runner/_work/intel-extension-for-openxla/intel-extension-for-openxla/jax
#UT results check
if [ -f "./logs/ut/err.log" ]; then
echo "exist fail cases"
Expand Down
Loading