Skip to content

Commit

Permalink
Add SDE Test (pytorch#327)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#327

Add SDE Test for Github action

Using "make -j2" instead of "make -j" to parallelize the build process. Using "make -j" seems not working on ubuntu here.

Differential Revision: D20590270

fbshipit-source-id: fef19fc660487d2234fdfe727aae6146da9cd23c
  • Loading branch information
Hongzhang Shan authored and facebook-github-bot committed Jun 14, 2020
1 parent 7d67304 commit 3bcd6a7
Showing 1 changed file with 37 additions and 2 deletions.
39 changes: 37 additions & 2 deletions .github/workflows/fbgemmci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,42 @@ jobs:
mkdir build_static
cd build_static
cmake -DUSE_SANITIZER=address -DFBGEMM_LIBRARY_TYPE=static ..
make -j
make -j2
- name: Download SDE
if: contains(runner.os, 'linux')
run: |
set -e
cd build_static
git clone https://github.com/marehr/intel-sde-downloader.git
pip install -r intel-sde-downloader/requirements.txt
python intel-sde-downloader/intel-sde-downloader.py sde-external-8.50.0-2020-03-26-lin.tar.bz2
tar -xf sde-external-8.50.0-2020-03-26-lin.tar.bz2
- name: Test SDE
if: contains(runner.os, 'linux')
run: |
set +e
cd build_static
sde-external-8.50.0-2020-03-26-lin/sde64 -icx -- test/Bfloat16ConvertTest
timeout -k 200s 180s sde-external-8.50.0-2020-03-26-lin/sde64 -icx -- test/EmbeddingSpMDM8BitTest
if [[ $? -eq 124 ]] || [[ $? -eq 137 ]]
then
echo "SDE EmbeddingSpMDM8BitTest Killed with return status: $?"
fi
timeout -k 140s 120s sde-external-8.50.0-2020-03-26-lin/sde64 -icx -- test/RequantizeOnlyTest
if [[ $? -eq 124 ]] || [[ $? -eq 137 ]]
then
echo "SDE RequantizeOnlyTest Killed with return status: $?"
fi
timeout -k 140s 120s sde-external-8.50.0-2020-03-26-lin/sde64 -icx -- test/RowWiseSparseAdagradFusedTest
if [[ $? -eq 124 ]] || [[ $? -eq 137 ]]
then
echo "SDE RowWiseSparseAdagradFusedTest Killed with return status: $?"
fi
echo "Finished Test SDE"
- name: Test static FBGEMM lib
if: contains(runner.os, 'linux') # not run on macos-latest now due to supporting AVX2
Expand All @@ -59,7 +94,7 @@ jobs:
mkdir build_shared
cd build_shared
cmake -DUSE_SANITIZER=address -DFBGEMM_LIBRARY_TYPE=shared ..
make -j
make -j2
- name: Test shared FBGEMM lib
if: contains(runner.os, 'linux') # not run on macos-latest now due to supporting AVX2
Expand Down

0 comments on commit 3bcd6a7

Please sign in to comment.