xgboost/tests/buildkite/conftest.sh
Bobby Wang 4e12f3e1bc
[Breaking][jvm-packages] Bump rapids version to 22.12.0 (#8648)
* [jvm-packages] Bump rapids version to 22.12.0

This PR bumps spark version to 3.1.1 and the rapids version
to 22.12.0, which results in the latest xgboost can't run
with the old rapids packages.
2023-01-07 18:59:17 +08:00

61 lines
1.5 KiB
Bash
Executable File

#!/bin/bash
set -euo pipefail
function get_aws_secret {
if [[ $# -ne 1 ]]
then
echo "Usage: get_aws_secret [Name of secret]"
return 1
fi
aws secretsmanager get-secret-value --secret-id $1 --output text --region us-west-2 --query SecretString
}
function set_buildkite_env_vars_in_container {
# Pass all Buildkite-specific env vars to Docker containers.
# This is to be used with tests/ci_build/ci_build.sh
export CI_DOCKER_EXTRA_PARAMS_INIT="${CI_DOCKER_EXTRA_PARAMS_INIT:-} "`
`"--env BUILDKITE_ANALYTICS_TOKEN --env BUILDKITE_BUILD_ID --env BUILDKITE_BUILD_NUMBER "`
`"--env BUILDKITE_JOB_ID --env BUILDKITE_BRANCH --env BUILDKITE_COMMIT "`
`"--env BUILDKITE_MESSAGE --env BUILDKITE_BUILD_URL"
}
set -x
CUDA_VERSION=11.0.3
RAPIDS_VERSION=22.10
SPARK_VERSION=3.1.1
JDK_VERSION=8
if [[ -z ${BUILDKITE:-} ]]
then
echo "$0 is not meant to run locally; it should run inside BuildKite."
echo "Please inspect the content of $0 and locate the desired command manually."
exit 1
fi
if [[ -n $BUILDKITE_PULL_REQUEST && $BUILDKITE_PULL_REQUEST != "false" ]]
then
is_pull_request=1
export BRANCH_NAME=PR-$BUILDKITE_PULL_REQUEST
else
is_pull_request=0
export BRANCH_NAME=$BUILDKITE_BRANCH
fi
if [[ $BUILDKITE_BRANCH == "master" || $BUILDKITE_BRANCH == "release_"* ]]
then
is_release_branch=1
enforce_daily_budget=0
else
is_release_branch=0
enforce_daily_budget=1
fi
if [[ -n ${DISABLE_RELEASE:-} ]]
then
is_release_branch=0
fi
set +x