silicon/design: add data disk

proppy · Oct 18, 2022 · 12eac53 · 12eac53
1 parent 292f556
commit 12eac53
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/modules/silicon_design/main.tf b/modules/silicon_design/main.tf
@@ -279,6 +279,9 @@ resource "google_notebooks_instance" "ai_notebook" {
   install_gpu_driver = false
   boot_disk_type     = var.boot_disk_type
   boot_disk_size_gb  = var.boot_disk_size_gb
+  data_disk_type     = var.data_disk_type
+  data_disk_size_gb  = var.data_disk_size_gb
+  no_remove_data_disk = true
 
   no_public_ip    = false
   no_proxy_access = false

diff --git a/modules/silicon_design/variables.tf b/modules/silicon_design/variables.tf
@@ -104,6 +104,18 @@ variable "create_budget" {
   default     = false
 }
 
+variable "data_disk_size_gb" {
+  description = "The size of the data disk in GB attached to this instance"
+  type        = number
+  default     = 1000
+}
+
+variable "data_disk_type" {
+  description = "Data disk types for notebook instances"
+  type        = string
+  default     = "PD_EXTREME"
+}
+
 variable "create_network" {
   description = "If the module has to be deployed in an existing network, set this variable to false. {{UIMeta group=2 order=1 }}"
   type        = bool