mirror of
https://github.com/tracel-ai/burn.git
synced 2026-05-31 19:49:48 +09:00
83 lines
1.8 KiB
TOML
83 lines
1.8 KiB
TOML
[environment]
|
|
gcp_gpu_attached = true
|
|
gcp_image_family = "tracel-ci-ubuntu-2404-amd64-nvidia"
|
|
# https://cloud.google.com/compute/docs/accelerator-optimized-machines
|
|
# put the faster machine on first place for possibly faster 'Benchmarks Started' feedback in PRs
|
|
gcp_machine_types = [
|
|
"a2-highgpu-1g", # 1 A100 40GB (listed as a2 standard)
|
|
"g2-standard-4", # 1 L4 24GB
|
|
]
|
|
# define the available zones for each machine type
|
|
# be sure to check what machine types are available in each region
|
|
# https://cloud.google.com/compute/docs/gpus/gpu-regions-zones#view-using-table
|
|
gcp_zones = [
|
|
# a2-highgpu-1g
|
|
[
|
|
"asia-northeast1-a",
|
|
"asia-northeast1-c",
|
|
"asia-northeast3-b",
|
|
"asia-southeast1-b",
|
|
"asia-southeast1-c",
|
|
"europe-west4-a",
|
|
"europe-west4-b",
|
|
"us-central1-a",
|
|
"us-central1-b",
|
|
"us-central1-c",
|
|
"us-central1-f",
|
|
"us-east1-b",
|
|
"us-west1-b",
|
|
"us-west3-b",
|
|
"us-west4-b"
|
|
],
|
|
# g2-standard-4
|
|
[
|
|
"northamerica-northeast2-a",
|
|
"northamerica-northeast2-b",
|
|
"us-central1-a",
|
|
"us-central1-b",
|
|
"us-central1-c",
|
|
"us-east1-b",
|
|
"us-east1-c",
|
|
"us-east1-d",
|
|
"us-east4-a",
|
|
"us-east4-c",
|
|
"us-west1-a",
|
|
"us-west1-b",
|
|
"us-west1-c",
|
|
"us-west4-a",
|
|
"us-west4-c"
|
|
],
|
|
]
|
|
repo_full = "tracel-ai/burn"
|
|
rust_toolchain = "stable"
|
|
rust_version = "stable"
|
|
|
|
[burn-bench]
|
|
github_organization = "tracel-ai"
|
|
github_repository = "burn-bench"
|
|
github_branch = "main"
|
|
github_workflow = "benchmarks.yml"
|
|
# vulkan autotune seems to take ages, disabling it for now
|
|
# backends = ["cuda-fusion", "vulkan-fusion", "wgpu-fusion"]
|
|
backends = ["cuda-fusion", "cuda"]
|
|
benches = ["autodiff",
|
|
"binary",
|
|
"bool_select",
|
|
"conv-transpose2d",
|
|
"conv-transpose3d",
|
|
"conv2d",
|
|
"conv3d",
|
|
"custom-gelu",
|
|
"data",
|
|
"load-record",
|
|
"matmul-fused",
|
|
"matmul",
|
|
"max-pool2d",
|
|
"random",
|
|
"reduce",
|
|
"softmax",
|
|
"transformer-encoder",
|
|
"unary"
|
|
]
|
|
dtypes = ["f16"]
|