kovagoadi/hunfabric
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for GPUs in Cloud Run

Browse Source
This commit is contained in:
Wiktor Niesiobędzki
2025-08-12 18:43:27 +00:00
parent 8468fd4ebc
commit e498f3407e
9 changed files with 379 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
modules/cloud-run-v2/workerpool.tf
View File

@@ -39,6 +39,13 @@ resource "google_cloud_run_v2_worker_pool" "default_managed" {
encryption_key = var.encryption_key
revision = local.revision_name
gpu_zonal_redundancy_disabled = var.revision.gpu_zonal_redundancy_disabled
dynamic "node_selector" {
for_each = var.revision.node_selector == null ? [] : [""]
content {
accelerator = var.revision.node_selector.accelerator
}
}
# Serverless VPC connector is not supported
# dynamic "vpc_access" {
# for_each = local.connector == null ? [] : [""]
@@ -178,7 +185,7 @@ resource "google_cloud_run_v2_worker_pool" "default_managed" {
}
resource "google_cloud_run_v2_worker_pool" "default_unmanaged" {
count = var.type == "WORKERPOOL" && var.managed_revision ? 1 : 0
count = var.type == "WORKERPOOL" && !var.managed_revision ? 1 : 0
provider = google-beta
project = var.project_id
location = var.region
@@ -202,6 +209,14 @@ resource "google_cloud_run_v2_worker_pool" "default_unmanaged" {
encryption_key = var.encryption_key
revision = local.revision_name
gpu_zonal_redundancy_disabled = var.revision.gpu_zonal_redundancy_disabled
dynamic "node_selector" {
for_each = var.revision.node_selector == null ? [] : [""]
content {
accelerator = var.revision.node_selector.accelerator
}
}
# Serverless VPC connector is not supported
# dynamic "vpc_access" {
# for_each = local.connector == null ? [] : [""]