list exo processes before test, warmup req in bench
This commit is contained in:
+10
-1
@@ -97,9 +97,18 @@ async def main() -> None:
|
||||
api_endpoint = "http://localhost:52415/v1/chat/completions"
|
||||
|
||||
# Define prompts
|
||||
prompt_warmup = "what is the capital of France?"
|
||||
prompt_essay = "write an essay about cats"
|
||||
|
||||
# Measure performance for the essay prompt, which depends on the first measurement
|
||||
# Warmup request
|
||||
print("\nPerforming warmup request...", flush=True)
|
||||
try:
|
||||
warmup_results = await measure_performance(api_endpoint, prompt_warmup)
|
||||
print("Warmup completed successfully", flush=True)
|
||||
except Exception as e:
|
||||
print(f"Warmup request failed: {e}", flush=True)
|
||||
|
||||
# Measure performance for the essay prompt
|
||||
print("\nMeasuring performance for the essay prompt...", flush=True)
|
||||
results = await measure_performance(api_endpoint, prompt_essay)
|
||||
|
||||
|
||||
@@ -57,6 +57,10 @@ jobs:
|
||||
aws_access_key_id: ${{ secrets.S3_EXO_BENCHMARKS_AWS_ACCESS_KEY_ID }}
|
||||
aws_secret_key: ${{ secrets.S3_EXO_BENCHMARKS_AWS_SECRET_ACCESS_KEY }}
|
||||
run: |
|
||||
# List existing exo processes
|
||||
echo "Existing exo processes:"
|
||||
ps aux | grep exo || true
|
||||
|
||||
CALLING_JOB="${{ inputs.calling_job_name }}"
|
||||
UNIQUE_JOB_ID="${CALLING_JOB}_${GITHUB_RUN_ID}"
|
||||
ALL_NODE_IDS=$(for i in $(seq ${{ strategy.job-total }} -1 0); do echo -n "${UNIQUE_JOB_ID}_${i},"; done | sed 's/,$//')
|
||||
|
||||
Reference in New Issue
Block a user