public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-08-05 16:48:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add ROCm Dockerfile to docker-bake.hcl

Browse source
This commit is contained in:
Kishor Prins 2025-06-29 10:49:34 -07:00
parent a109a335a4
commit ce3cd62ee5
1 changed files with 25 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
docker-bake.hcl
View file

@ -40,6 +40,12 @@ target "_gpu_base" {
dockerfile = "docker/gpu/Dockerfile"
}
# Base settings for AMD ROCm builds
target "_rocm_base" {
inherits = ["_common"]
dockerfile = "docker/rocm/Dockerfile"
}
# CPU target with multi-platform support
target "cpu" {
inherits = ["_cpu_base"]
@ -60,9 +66,19 @@ target "gpu" {
]
}
# AMD ROCm target with multi-platform support
target "rocm" {
inherits = ["_rocm_base"]
platforms = ["linux/amd64", "linux/arm64"]
tags = [
"${REGISTRY}/${OWNER}/${REPO}-rocm:${VERSION}",
"${REGISTRY}/${OWNER}/${REPO}-rocm:latest"
]
}
# Default group to build both CPU and GPU versions
group "default" {
targets = ["cpu", "gpu"]
targets = ["cpu", "gpu", "rocm"]
}
# Development targets for faster local builds
@ -78,6 +94,12 @@ target "gpu-dev" {
tags = ["${REGISTRY}/${OWNER}/${REPO}-gpu:dev"]
}
target "rocm-dev" {
inherits = ["_rocm_base"]
# No multi-platform for dev builds
tags = ["${REGISTRY}/${OWNER}/${REPO}-rocm:dev"]
}
group "dev" {
targets = ["cpu-dev", "gpu-dev"]
}
targets = ["cpu-dev", "gpu-dev", "rocm-dev"]
}