Unverified Commit 9e897181 authored by Dmitry Tokarev's avatar Dmitry Tokarev Committed by GitHub
Browse files

fix: Fix pre-merge (#6341)


Signed-off-by: default avatarDmitry Tokarev <dtokarev@nvidia.com>
parent ea02149e
...@@ -87,6 +87,7 @@ jobs: ...@@ -87,6 +87,7 @@ jobs:
target: frontend target: frontend
platform: ${{ env.PLATFORMS }} platform: ${{ env.PLATFORMS }}
ci_token: ${{ secrets.CI_TOKEN }} ci_token: ${{ secrets.CI_TOKEN }}
cuda_version: "12.9"
aws_default_region: ${{ secrets.AWS_DEFAULT_REGION }} aws_default_region: ${{ secrets.AWS_DEFAULT_REGION }}
sccache_s3_bucket: ${{ secrets.SCCACHE_S3_BUCKET }} sccache_s3_bucket: ${{ secrets.SCCACHE_S3_BUCKET }}
aws_account_id: ${{ secrets.AWS_ACCOUNT_ID }} aws_account_id: ${{ secrets.AWS_ACCOUNT_ID }}
......
...@@ -58,11 +58,25 @@ def parse_args(): ...@@ -58,11 +58,25 @@ def parse_args():
def validate_args(args): def validate_args(args):
valid_inputs = { valid_inputs = {
"vllm": { "vllm": {
"target": ["runtime", "dev", "local-dev", "framework", "wheel_builder", "base"], "target": [
"runtime",
"dev",
"local-dev",
"framework",
"wheel_builder",
"base",
],
"cuda_version": ["12.9", "13.0"], "cuda_version": ["12.9", "13.0"],
}, },
"trtllm": { "trtllm": {
"target": ["runtime", "dev", "local-dev", "framework", "wheel_builder", "base"], "target": [
"runtime",
"dev",
"local-dev",
"framework",
"wheel_builder",
"base",
],
"cuda_version": ["13.1"], "cuda_version": ["13.1"],
}, },
"sglang": { "sglang": {
...@@ -70,13 +84,23 @@ def validate_args(args): ...@@ -70,13 +84,23 @@ def validate_args(args):
"cuda_version": ["12.9", "13.0"], "cuda_version": ["12.9", "13.0"],
}, },
"dynamo": { "dynamo": {
"target": ["runtime", "dev", "local-dev", "frontend", "wheel_builder", "base"], "target": [
"runtime",
"dev",
"local-dev",
"frontend",
"wheel_builder",
"base",
],
"cuda_version": ["12.9", "13.0"], "cuda_version": ["12.9", "13.0"],
}, },
} }
if args.framework in valid_inputs: if args.framework in valid_inputs:
if args.target in valid_inputs[args.framework]["target"] and args.cuda_version in valid_inputs[args.framework]["cuda_version"]: if (
args.target in valid_inputs[args.framework]["target"]
and args.cuda_version in valid_inputs[args.framework]["cuda_version"]
):
return return
else: else:
raise ValueError( raise ValueError(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment