Skip to content

Commit 3a63713

Browse files
authored
Add an acceptance test for ssh connection to serverless GPU clusters (#4579)
## Changes <!-- Brief summary of your changes that is easy to understand --> Introduce a new cloud test that makes a simple connection to a serverless gpu compute with "echo 'Hello'" and check that databricks ssh connect output has it.
1 parent af75ba1 commit 3a63713

File tree

5 files changed

+28
-0
lines changed

5 files changed

+28
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Connection successful

acceptance/ssh/connect-serverless-gpu/out.test.toml

Lines changed: 9 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

acceptance/ssh/connect-serverless-gpu/output.txt

Whitespace-only changes.
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
errcode $CLI ssh connect --name serverless-gpu-test --accelerator=GPU_1xA10 --releases-dir=$CLI_RELEASES_DIR -- "echo 'Connection successful'" >out.stdout.txt 2>LOG.stderr
2+
3+
if ! grep -q "Connection successful" out.stdout.txt; then
4+
run_id=$(cat LOG.stderr | grep -o "Job submitted successfully with run ID: [0-9]*" | grep -o "[0-9]*$")
5+
trace $CLI jobs get-run "$run_id" > LOG.job
6+
fi
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
Local = false
2+
Cloud = true
3+
4+
# Serverless GPU is only available in newer environments
5+
RequiresUnityCatalog = true
6+
7+
# Serverless GPU is not available in GCP yet
8+
[CloudEnvs]
9+
gcp = false
10+
11+
[EnvMatrix]
12+
DATABRICKS_BUNDLE_ENGINE = ["direct"]

0 commit comments

Comments
 (0)