Skip to content

Commit

Permalink
Update check_binary.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
andysamfb committed Sep 20, 2022
1 parent 051b9d1 commit 6850bed
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion check_binary.sh
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,6 @@ setup_link_flags () {
}

TEST_CODE_DIR="$(dirname ${BASH_SOURCE[0]})/test_example_code"
echo ${TEST_CODE_DIR}
build_and_run_example_cpp () {
if [[ "$DESIRED_DEVTOOLSET" == *"cxx11-abi"* ]]; then
GLIBCXX_USE_CXX11_ABI=1
Expand Down Expand Up @@ -386,6 +385,8 @@ if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != *"rocm"* ]]; then
# Validates builds is free of linker regressions reported in https://github.com/pytorch/pytorch/issues/57744
echo "Checking that exception handling works"
python -c "import torch; from unittest import TestCase;TestCase().assertRaises(RuntimeError, lambda:torch.eye(7, 7, device='cuda:7'))"

echo "test_code_dir = ${TEST_CODE_DIR}"

echo "Checking that basic RNN works"
python ${TEST_CODE_DIR}/rnn_smoke.py
Expand Down

0 comments on commit 6850bed

Please sign in to comment.