Skip to content

Commit

Permalink
Enable all tests for arm builds
Browse files Browse the repository at this point in the history
  • Loading branch information
galipremsagar committed Mar 27, 2024
1 parent a7ceede commit aa805b5
Showing 1 changed file with 9 additions and 15 deletions.
24 changes: 9 additions & 15 deletions ci/test_wheel_cudf.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,18 +13,12 @@ RESULTS_DIR=${RAPIDS_TESTS_DIR:-"$(mktemp -d)"}
RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${RESULTS_DIR}/test-results"}/
mkdir -p "${RAPIDS_TESTS_DIR}"

# Run smoke tests for aarch64 pull requests
if [[ "$(arch)" == "aarch64" && ${RAPIDS_BUILD_TYPE} == "pull-request" ]]; then
rapids-logger "Run smoke tests for cudf"
python ./ci/wheel_smoke_test_cudf.py
else
rapids-logger "pytest cudf"
pushd python/cudf/cudf/tests
python -m pytest \
--cache-clear \
--junitxml="${RAPIDS_TESTS_DIR}/junit-cudf.xml" \
--numprocesses=8 \
--dist=worksteal \
.
popd
fi
rapids-logger "pytest cudf"
pushd python/cudf/cudf/tests
python -m pytest \
--cache-clear \
--junitxml="${RAPIDS_TESTS_DIR}/junit-cudf.xml" \
--numprocesses=8 \
--dist=worksteal \
.
popd

0 comments on commit aa805b5

Please sign in to comment.