diff --git a/google/provider.go b/google/provider.go index 2eee610d..3cd1661a 100644 --- a/google/provider.go +++ b/google/provider.go @@ -145,6 +145,7 @@ func ResourceMapWithErrors() (map[string]*schema.Resource, error) { GeneratedSourceRepoResourcesMap, GeneratedSpannerResourcesMap, GeneratedStorageResourcesMap, + GeneratedTpuResourcesMap, GeneratedMonitoringResourcesMap, map[string]*schema.Resource{ "google_app_engine_application": resourceAppEngineApplication(), diff --git a/google/provider_tpu_gen.go b/google/provider_tpu_gen.go new file mode 100644 index 00000000..f2690560 --- /dev/null +++ b/google/provider_tpu_gen.go @@ -0,0 +1,21 @@ +// ---------------------------------------------------------------------------- +// +// *** AUTO GENERATED CODE *** AUTO GENERATED CODE *** +// +// ---------------------------------------------------------------------------- +// +// This file is automatically generated by Magic Modules and manual +// changes will be clobbered when the file is regenerated. +// +// Please read more about how to change this file in +// .github/CONTRIBUTING.md. +// +// ---------------------------------------------------------------------------- + +package google + +import "github.com/hashicorp/terraform/helper/schema" + +var GeneratedTpuResourcesMap = map[string]*schema.Resource{ + "google_tpu_node": resourceTpuNode(), +} diff --git a/google/resource_tpu_node.go b/google/resource_tpu_node.go new file mode 100644 index 00000000..f8b959cf --- /dev/null +++ b/google/resource_tpu_node.go @@ -0,0 +1,527 @@ +// ---------------------------------------------------------------------------- +// +// *** AUTO GENERATED CODE *** AUTO GENERATED CODE *** +// +// ---------------------------------------------------------------------------- +// +// This file is automatically generated by Magic Modules and manual +// changes will be clobbered when the file is regenerated. +// +// Please read more about how to change this file in +// .github/CONTRIBUTING.md. +// +// ---------------------------------------------------------------------------- + +package google + +import ( + "fmt" + "log" + "reflect" + "strconv" + "time" + + "github.com/hashicorp/terraform/helper/schema" +) + +// compareTpuNodeSchedulingConfig diff suppresses for the default +// scheduling, i.e. if preemptible is false, the API may either return no +// schedulingConfig or an empty schedulingConfig. +func compareTpuNodeSchedulingConfig(k, old, new string, d *schema.ResourceData) bool { + if k == "scheduling_config.0.preemptible" { + return old == "" && new == "false" + } + if k == "scheduling_config.#" { + o, n := d.GetChange("scheduling_config.0.preemptible") + return o.(bool) == n.(bool) + } + return false +} + +func resourceTpuNode() *schema.Resource { + return &schema.Resource{ + Create: resourceTpuNodeCreate, + Read: resourceTpuNodeRead, + Update: resourceTpuNodeUpdate, + Delete: resourceTpuNodeDelete, + + Importer: &schema.ResourceImporter{ + State: resourceTpuNodeImport, + }, + + Timeouts: &schema.ResourceTimeout{ + Create: schema.DefaultTimeout(900 * time.Second), + Update: schema.DefaultTimeout(900 * time.Second), + Delete: schema.DefaultTimeout(900 * time.Second), + }, + + Schema: map[string]*schema.Schema{ + "accelerator_type": { + Type: schema.TypeString, + Required: true, + ForceNew: true, + }, + "cidr_block": { + Type: schema.TypeString, + Required: true, + ForceNew: true, + }, + "name": { + Type: schema.TypeString, + Required: true, + ForceNew: true, + }, + "tensorflow_version": { + Type: schema.TypeString, + Required: true, + }, + "zone": { + Type: schema.TypeString, + Required: true, + ForceNew: true, + }, + "description": { + Type: schema.TypeString, + Optional: true, + ForceNew: true, + }, + "labels": { + Type: schema.TypeMap, + Optional: true, + ForceNew: true, + Elem: &schema.Schema{Type: schema.TypeString}, + }, + "network": { + Type: schema.TypeString, + Computed: true, + Optional: true, + ForceNew: true, + DiffSuppressFunc: compareSelfLinkOrResourceName, + }, + "scheduling_config": { + Type: schema.TypeList, + Optional: true, + ForceNew: true, + DiffSuppressFunc: compareTpuNodeSchedulingConfig, + MaxItems: 1, + Elem: &schema.Resource{ + Schema: map[string]*schema.Schema{ + "preemptible": { + Type: schema.TypeBool, + Optional: true, + ForceNew: true, + DiffSuppressFunc: compareTpuNodeSchedulingConfig, + Default: false, + }, + }, + }, + }, + "network_endpoints": { + Type: schema.TypeList, + Computed: true, + Elem: &schema.Resource{ + Schema: map[string]*schema.Schema{ + "ip_address": { + Type: schema.TypeString, + Computed: true, + }, + "port": { + Type: schema.TypeInt, + Computed: true, + }, + }, + }, + }, + "service_account": { + Type: schema.TypeString, + Computed: true, + }, + "project": { + Type: schema.TypeString, + Optional: true, + Computed: true, + ForceNew: true, + }, + }, + } +} + +func resourceTpuNodeCreate(d *schema.ResourceData, meta interface{}) error { + config := meta.(*Config) + + obj := make(map[string]interface{}) + nameProp, err := expandTpuNodeName(d.Get("name"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("name"); !isEmptyValue(reflect.ValueOf(nameProp)) && (ok || !reflect.DeepEqual(v, nameProp)) { + obj["name"] = nameProp + } + descriptionProp, err := expandTpuNodeDescription(d.Get("description"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("description"); !isEmptyValue(reflect.ValueOf(descriptionProp)) && (ok || !reflect.DeepEqual(v, descriptionProp)) { + obj["description"] = descriptionProp + } + acceleratorTypeProp, err := expandTpuNodeAcceleratorType(d.Get("accelerator_type"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("accelerator_type"); !isEmptyValue(reflect.ValueOf(acceleratorTypeProp)) && (ok || !reflect.DeepEqual(v, acceleratorTypeProp)) { + obj["acceleratorType"] = acceleratorTypeProp + } + tensorflowVersionProp, err := expandTpuNodeTensorflowVersion(d.Get("tensorflow_version"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("tensorflow_version"); !isEmptyValue(reflect.ValueOf(tensorflowVersionProp)) && (ok || !reflect.DeepEqual(v, tensorflowVersionProp)) { + obj["tensorflowVersion"] = tensorflowVersionProp + } + networkProp, err := expandTpuNodeNetwork(d.Get("network"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("network"); !isEmptyValue(reflect.ValueOf(networkProp)) && (ok || !reflect.DeepEqual(v, networkProp)) { + obj["network"] = networkProp + } + cidrBlockProp, err := expandTpuNodeCidrBlock(d.Get("cidr_block"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("cidr_block"); !isEmptyValue(reflect.ValueOf(cidrBlockProp)) && (ok || !reflect.DeepEqual(v, cidrBlockProp)) { + obj["cidrBlock"] = cidrBlockProp + } + schedulingConfigProp, err := expandTpuNodeSchedulingConfig(d.Get("scheduling_config"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("scheduling_config"); !isEmptyValue(reflect.ValueOf(schedulingConfigProp)) && (ok || !reflect.DeepEqual(v, schedulingConfigProp)) { + obj["schedulingConfig"] = schedulingConfigProp + } + labelsProp, err := expandTpuNodeLabels(d.Get("labels"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("labels"); !isEmptyValue(reflect.ValueOf(labelsProp)) && (ok || !reflect.DeepEqual(v, labelsProp)) { + obj["labels"] = labelsProp + } + + url, err := replaceVars(d, config, "https://tpu.googleapis.com/v1/projects/{{project}}/locations/{{zone}}/nodes?nodeId={{name}}") + if err != nil { + return err + } + + log.Printf("[DEBUG] Creating new Node: %#v", obj) + res, err := sendRequestWithTimeout(config, "POST", url, obj, d.Timeout(schema.TimeoutCreate)) + if err != nil { + return fmt.Errorf("Error creating Node: %s", err) + } + + // Store the ID now + id, err := replaceVars(d, config, "{{project}}/{{zone}}/{{name}}") + if err != nil { + return fmt.Errorf("Error constructing id: %s", err) + } + d.SetId(id) + + project, err := getProject(d, config) + if err != nil { + return err + } + waitErr := tpuOperationWaitTime( + config, res, project, "Creating Node", + int(d.Timeout(schema.TimeoutCreate).Minutes())) + + if waitErr != nil { + // The resource didn't actually create + d.SetId("") + return fmt.Errorf("Error waiting to create Node: %s", waitErr) + } + + log.Printf("[DEBUG] Finished creating Node %q: %#v", d.Id(), res) + + return resourceTpuNodeRead(d, meta) +} + +func resourceTpuNodeRead(d *schema.ResourceData, meta interface{}) error { + config := meta.(*Config) + + url, err := replaceVars(d, config, "https://tpu.googleapis.com/v1/projects/{{project}}/locations/{{zone}}/nodes/{{name}}") + if err != nil { + return err + } + + res, err := sendRequest(config, "GET", url, nil) + if err != nil { + return handleNotFoundError(err, d, fmt.Sprintf("TpuNode %q", d.Id())) + } + + project, err := getProject(d, config) + if err != nil { + return err + } + if err := d.Set("project", project); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + + if err := d.Set("name", flattenTpuNodeName(res["name"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("description", flattenTpuNodeDescription(res["description"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("accelerator_type", flattenTpuNodeAcceleratorType(res["acceleratorType"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("tensorflow_version", flattenTpuNodeTensorflowVersion(res["tensorflowVersion"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("network", flattenTpuNodeNetwork(res["network"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("cidr_block", flattenTpuNodeCidrBlock(res["cidrBlock"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("service_account", flattenTpuNodeServiceAccount(res["serviceAccount"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("scheduling_config", flattenTpuNodeSchedulingConfig(res["schedulingConfig"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("network_endpoints", flattenTpuNodeNetworkEndpoints(res["networkEndpoints"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + if err := d.Set("labels", flattenTpuNodeLabels(res["labels"], d)); err != nil { + return fmt.Errorf("Error reading Node: %s", err) + } + + return nil +} + +func resourceTpuNodeUpdate(d *schema.ResourceData, meta interface{}) error { + config := meta.(*Config) + + d.Partial(true) + + if d.HasChange("tensorflow_version") { + obj := make(map[string]interface{}) + tensorflowVersionProp, err := expandTpuNodeTensorflowVersion(d.Get("tensorflow_version"), d, config) + if err != nil { + return err + } else if v, ok := d.GetOkExists("tensorflow_version"); !isEmptyValue(reflect.ValueOf(v)) && (ok || !reflect.DeepEqual(v, tensorflowVersionProp)) { + obj["tensorflowVersion"] = tensorflowVersionProp + } + + url, err := replaceVars(d, config, "https://tpu.googleapis.com/v1/projects/{{project}}/locations/{{zone}}/nodes/{{name}}:reimage") + if err != nil { + return err + } + res, err := sendRequestWithTimeout(config, "POST", url, obj, d.Timeout(schema.TimeoutUpdate)) + if err != nil { + return fmt.Errorf("Error updating Node %q: %s", d.Id(), err) + } + + project, err := getProject(d, config) + if err != nil { + return err + } + + err = tpuOperationWaitTime( + config, res, project, "Updating Node", + int(d.Timeout(schema.TimeoutUpdate).Minutes())) + + if err != nil { + return err + } + + d.SetPartial("tensorflow_version") + } + + d.Partial(false) + + return resourceTpuNodeRead(d, meta) +} + +func resourceTpuNodeDelete(d *schema.ResourceData, meta interface{}) error { + config := meta.(*Config) + + url, err := replaceVars(d, config, "https://tpu.googleapis.com/v1/projects/{{project}}/locations/{{zone}}/nodes/{{name}}") + if err != nil { + return err + } + + var obj map[string]interface{} + log.Printf("[DEBUG] Deleting Node %q", d.Id()) + res, err := sendRequestWithTimeout(config, "DELETE", url, obj, d.Timeout(schema.TimeoutDelete)) + if err != nil { + return handleNotFoundError(err, d, "Node") + } + + project, err := getProject(d, config) + if err != nil { + return err + } + + err = tpuOperationWaitTime( + config, res, project, "Deleting Node", + int(d.Timeout(schema.TimeoutDelete).Minutes())) + + if err != nil { + return err + } + + log.Printf("[DEBUG] Finished deleting Node %q: %#v", d.Id(), res) + return nil +} + +func resourceTpuNodeImport(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) { + config := meta.(*Config) + if err := parseImportId([]string{"projects/(?P[^/]+)/locations/(?P[^/]+)/nodes/(?P[^/]+)", "(?P[^/]+)/(?P[^/]+)/(?P[^/]+)", "(?P[^/]+)"}, d, config); err != nil { + return nil, err + } + + // Replace import id for the resource id + id, err := replaceVars(d, config, "{{project}}/{{zone}}/{{name}}") + if err != nil { + return nil, fmt.Errorf("Error constructing id: %s", err) + } + d.SetId(id) + + return []*schema.ResourceData{d}, nil +} + +func flattenTpuNodeName(v interface{}, d *schema.ResourceData) interface{} { + if v == nil { + return v + } + return NameFromSelfLinkStateFunc(v) +} + +func flattenTpuNodeDescription(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeAcceleratorType(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeTensorflowVersion(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeNetwork(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeCidrBlock(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeServiceAccount(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeSchedulingConfig(v interface{}, d *schema.ResourceData) interface{} { + if v == nil { + return nil + } + original := v.(map[string]interface{}) + if len(original) == 0 { + return nil + } + transformed := make(map[string]interface{}) + transformed["preemptible"] = + flattenTpuNodeSchedulingConfigPreemptible(original["preemptible"], d) + return []interface{}{transformed} +} +func flattenTpuNodeSchedulingConfigPreemptible(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeNetworkEndpoints(v interface{}, d *schema.ResourceData) interface{} { + if v == nil { + return v + } + l := v.([]interface{}) + transformed := make([]interface{}, 0, len(l)) + for _, raw := range l { + original := raw.(map[string]interface{}) + if len(original) < 1 { + // Do not include empty json objects coming back from the api + continue + } + transformed = append(transformed, map[string]interface{}{ + "ip_address": flattenTpuNodeNetworkEndpointsIpAddress(original["ipAddress"], d), + "port": flattenTpuNodeNetworkEndpointsPort(original["port"], d), + }) + } + return transformed +} +func flattenTpuNodeNetworkEndpointsIpAddress(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func flattenTpuNodeNetworkEndpointsPort(v interface{}, d *schema.ResourceData) interface{} { + // Handles the string fixed64 format + if strVal, ok := v.(string); ok { + if intVal, err := strconv.ParseInt(strVal, 10, 64); err == nil { + return intVal + } // let terraform core handle it if we can't convert the string to an int. + } + return v +} + +func flattenTpuNodeLabels(v interface{}, d *schema.ResourceData) interface{} { + return v +} + +func expandTpuNodeName(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeDescription(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeAcceleratorType(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeTensorflowVersion(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeNetwork(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeCidrBlock(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeSchedulingConfig(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + l := v.([]interface{}) + if len(l) == 0 || l[0] == nil { + return nil, nil + } + raw := l[0] + original := raw.(map[string]interface{}) + transformed := make(map[string]interface{}) + + transformedPreemptible, err := expandTpuNodeSchedulingConfigPreemptible(original["preemptible"], d, config) + if err != nil { + return nil, err + } else if val := reflect.ValueOf(transformedPreemptible); val.IsValid() && !isEmptyValue(val) { + transformed["preemptible"] = transformedPreemptible + } + + return transformed, nil +} + +func expandTpuNodeSchedulingConfigPreemptible(v interface{}, d TerraformResourceData, config *Config) (interface{}, error) { + return v, nil +} + +func expandTpuNodeLabels(v interface{}, d TerraformResourceData, config *Config) (map[string]string, error) { + if v == nil { + return map[string]string{}, nil + } + m := make(map[string]string) + for k, val := range v.(map[string]interface{}) { + m[k] = val.(string) + } + return m, nil +} diff --git a/google/resource_tpu_node_generated_test.go b/google/resource_tpu_node_generated_test.go new file mode 100644 index 00000000..5313e6b8 --- /dev/null +++ b/google/resource_tpu_node_generated_test.go @@ -0,0 +1,143 @@ +// ---------------------------------------------------------------------------- +// +// *** AUTO GENERATED CODE *** AUTO GENERATED CODE *** +// +// ---------------------------------------------------------------------------- +// +// This file is automatically generated by Magic Modules and manual +// changes will be clobbered when the file is regenerated. +// +// Please read more about how to change this file in +// .github/CONTRIBUTING.md. +// +// ---------------------------------------------------------------------------- + +package google + +import ( + "fmt" + "strings" + "testing" + + "github.com/hashicorp/terraform/helper/acctest" + "github.com/hashicorp/terraform/helper/resource" + "github.com/hashicorp/terraform/terraform" +) + +func TestAccTpuNode_tpuNodeBasicExample(t *testing.T) { + t.Parallel() + + context := map[string]interface{}{ + "random_suffix": acctest.RandString(10), + } + + resource.Test(t, resource.TestCase{ + PreCheck: func() { testAccPreCheck(t) }, + Providers: testAccProviders, + CheckDestroy: testAccCheckTpuNodeDestroy, + Steps: []resource.TestStep{ + { + Config: testAccTpuNode_tpuNodeBasicExample(context), + }, + { + ResourceName: "google_tpu_node.tpu", + ImportState: true, + ImportStateVerify: true, + ImportStateVerifyIgnore: []string{"zone"}, + }, + }, + }) +} + +func testAccTpuNode_tpuNodeBasicExample(context map[string]interface{}) string { + return Nprintf(` +resource "google_tpu_node" "tpu" { + name = "test-tpu-%{random_suffix}" + zone = "us-central1-b" + + accelerator_type = "v3-8" + tensorflow_version = "1.9" + cidr_block = "10.2.0.0/29" +} +`, context) +} + +func TestAccTpuNode_tpuNodeFullExample(t *testing.T) { + t.Parallel() + + context := map[string]interface{}{ + "random_suffix": acctest.RandString(10), + } + + resource.Test(t, resource.TestCase{ + PreCheck: func() { testAccPreCheck(t) }, + Providers: testAccProviders, + CheckDestroy: testAccCheckTpuNodeDestroy, + Steps: []resource.TestStep{ + { + Config: testAccTpuNode_tpuNodeFullExample(context), + }, + { + ResourceName: "google_tpu_node.tpu", + ImportState: true, + ImportStateVerify: true, + ImportStateVerifyIgnore: []string{"zone"}, + }, + }, + }) +} + +func testAccTpuNode_tpuNodeFullExample(context map[string]interface{}) string { + return Nprintf(` +resource "google_compute_network" "tpu_network" { + name = "test-tpu-network-%{random_suffix}" + auto_create_subnetworks = false +} + +resource "google_tpu_node" "tpu" { + name = "test-tpu-%{random_suffix}" + zone = "us-central1-b" + + accelerator_type = "v3-8" + + cidr_block = "10.3.0.0/29" + tensorflow_version = "1.12" + + description = "Terraform Google Provider test TPU" + network = "${google_compute_network.tpu_network.name}" + + labels { + foo = "bar" + } + + scheduling_config { + preemptible = true + } +} +`, context) +} + +func testAccCheckTpuNodeDestroy(s *terraform.State) error { + for name, rs := range s.RootModule().Resources { + if rs.Type != "google_tpu_node" { + continue + } + if strings.HasPrefix(name, "data.") { + continue + } + + config := testAccProvider.Meta().(*Config) + + url, err := replaceVarsForTest(rs, "https://tpu.googleapis.com/v1/projects/{{project}}/locations/{{zone}}/nodes/{{name}}") + if err != nil { + return err + } + + _, err = sendRequest(config, "GET", url, nil) + if err == nil { + return fmt.Errorf("TpuNode still exists at %s", url) + } + } + + return nil +} diff --git a/google/resource_tpu_node_test.go b/google/resource_tpu_node_test.go new file mode 100644 index 00000000..dd107b79 --- /dev/null +++ b/google/resource_tpu_node_test.go @@ -0,0 +1,57 @@ +package google + +import ( + "testing" + + "fmt" + "github.com/hashicorp/terraform/helper/acctest" + "github.com/hashicorp/terraform/helper/resource" +) + +func TestAccTpuNode_tpuNodeBUpdateTensorFlowVersion(t *testing.T) { + t.Parallel() + + nodeId := acctest.RandomWithPrefix("tf-test") + + resource.Test(t, resource.TestCase{ + PreCheck: func() { testAccPreCheck(t) }, + Providers: testAccProviders, + CheckDestroy: testAccCheckTpuNodeDestroy, + Steps: []resource.TestStep{ + { + Config: testAccTpuNode_tpuNodeTensorFlow(nodeId, "1.11"), + }, + { + ResourceName: "google_tpu_node.tpu", + ImportState: true, + ImportStateVerify: true, + ImportStateVerifyIgnore: []string{"zone"}, + }, + { + Config: testAccTpuNode_tpuNodeTensorFlow(nodeId, "1.12"), + }, + { + ResourceName: "google_tpu_node.tpu", + ImportState: true, + ImportStateVerify: true, + ImportStateVerifyIgnore: []string{"zone"}, + }, + }, + }) +} + +// WARNING: cidr_block must not overlap with other existing TPU blocks +// Make sure if you change this value that it does not overlap with the +// autogenerated examples. +func testAccTpuNode_tpuNodeTensorFlow(nodeId, tensorFlowVer string) string { + return fmt.Sprintf(` +resource "google_tpu_node" "tpu" { + name = "%s" + zone = "us-central1-b" + + accelerator_type = "v3-8" + tensorflow_version = "%s" + cidr_block = "10.1.0.0/29" +} +`, nodeId, tensorFlowVer) +} diff --git a/google/tpu_operation.go b/google/tpu_operation.go new file mode 100644 index 00000000..cab45b3f --- /dev/null +++ b/google/tpu_operation.go @@ -0,0 +1,46 @@ +// ---------------------------------------------------------------------------- +// +// *** AUTO GENERATED CODE *** AUTO GENERATED CODE *** +// +// ---------------------------------------------------------------------------- +// +// This file is automatically generated by Magic Modules and manual +// changes will be clobbered when the file is regenerated. +// +// Please read more about how to change this file in +// .github/CONTRIBUTING.md. +// +// ---------------------------------------------------------------------------- +package google + +import ( + "fmt" +) + +type TpuOperationWaiter struct { + Config *Config + CommonOperationWaiter +} + +func (w *TpuOperationWaiter) QueryOp() (interface{}, error) { + if w == nil { + return nil, fmt.Errorf("Cannot query operation, it's unset or nil.") + } + // Returns the proper get. + url := fmt.Sprintf("https://tpu.googleapis.com/v1/%s", w.CommonOperationWaiter.Op.Name) + return sendRequest(w.Config, "GET", url, nil) +} + +func tpuOperationWaitTime(config *Config, op map[string]interface{}, project, activity string, timeoutMinutes int) error { + if val, ok := op["name"]; !ok || val == "" { + // This was a synchronous call - there is no operation to wait for. + return nil + } + w := &TpuOperationWaiter{ + Config: config, + } + if err := w.CommonOperationWaiter.SetOp(op); err != nil { + return err + } + return OperationWait(w, activity, timeoutMinutes) +} diff --git a/website/docs/r/tpu_node.html.markdown b/website/docs/r/tpu_node.html.markdown new file mode 100644 index 00000000..9c46ee8a --- /dev/null +++ b/website/docs/r/tpu_node.html.markdown @@ -0,0 +1,196 @@ +--- +# ---------------------------------------------------------------------------- +# +# *** AUTO GENERATED CODE *** AUTO GENERATED CODE *** +# +# ---------------------------------------------------------------------------- +# +# This file is automatically generated by Magic Modules and manual +# changes will be clobbered when the file is regenerated. +# +# Please read more about how to change this file in +# .github/CONTRIBUTING.md. +# +# ---------------------------------------------------------------------------- +layout: "google" +page_title: "Google: google_tpu_node" +sidebar_current: "docs-google-tpu-node" +description: |- + A Cloud TPU instance. +--- + +# google\_tpu\_node + +A Cloud TPU instance. + + +To get more information about Node, see: + +* [API documentation](https://cloud.google.com/tpu/docs/reference/rest/) +* How-to Guides + * [Official Documentation](https://cloud.google.com/tpu/docs/) + +
+ + Open in Cloud Shell + +
+## Example Usage - Tpu Node Basic + + +```hcl +resource "google_tpu_node" "tpu" { + name = "test-tpu" + zone = "us-central1-b" + + accelerator_type = "v3-8" + tensorflow_version = "1.9" + cidr_block = "10.2.0.0/29" +} +``` +
+ + Open in Cloud Shell + +
+## Example Usage - Tpu Node Full + + +```hcl +resource "google_compute_network" "tpu_network" { + name = "test-tpu-network" + auto_create_subnetworks = false +} + +resource "google_tpu_node" "tpu" { + name = "test-tpu" + zone = "us-central1-b" + + accelerator_type = "v3-8" + + cidr_block = "10.3.0.0/29" + tensorflow_version = "1.12" + + description = "Terraform Google Provider test TPU" + network = "${google_compute_network.tpu_network.name}" + + labels { + foo = "bar" + } + + scheduling_config { + preemptible = true + } +} +``` + +## Argument Reference + +The following arguments are supported: + + +* `name` - + (Required) + The immutable name of the TPU. + +* `accelerator_type` - + (Required) + The type of hardware accelerators associated with this node. + +* `tensorflow_version` - + (Required) + The version of Tensorflow running in the Node. + +* `cidr_block` - + (Required) + The CIDR block that the TPU node will use when selecting an IP + address. This CIDR block must be a /29 block; the Compute Engine + networks API forbids a smaller block, and using a larger block would + be wasteful (a node can only consume one IP address). + Errors will occur if the CIDR block has already been used for a + currently existing TPU node, the CIDR block conflicts with any + subnetworks in the user's provided network, or the provided network + is peered with another network that is using that CIDR block. + +* `zone` - + (Required) + The GCP location for the TPU. + + +- - - + + +* `description` - + (Optional) + The user-supplied description of the TPU. Maximum of 512 characters. + +* `network` - + (Optional) + The name of a network to peer the TPU node to. It must be a + preexisting Compute Engine network inside of the project on which + this API has been activated. If none is provided, "default" will be + used. + +* `scheduling_config` - + (Optional) + Sets the scheduling options for this TPU instance. Structure is documented below. + +* `labels` - + (Optional) + Resource labels to represent user provided metadata. +* `project` - (Optional) The ID of the project in which the resource belongs. + If it is not provided, the provider project is used. + + +The `scheduling_config` block supports: + +* `preemptible` - + (Optional) + Defines whether the TPU instance is preemptible. + +## Attributes Reference + +In addition to the arguments listed above, the following computed attributes are exported: + + +* `service_account` - + The service account used to run the tensor flow services within the + node. To share resources, including Google Cloud Storage data, with + the Tensorflow job running in the Node, this account must have + permissions to that data. + +* `network_endpoints` - + The network endpoints where TPU workers can be accessed and sent work. + It is recommended that Tensorflow clients of the node first reach out + to the first (index 0) entry. Structure is documented below. + + +The `network_endpoints` block contains: + +* `ip_address` - + The IP address of this network endpoint. + +* `port` - + The port of this network endpoint. + +## Timeouts + +This resource provides the following +[Timeouts](/docs/configuration/resources.html#timeouts) configuration options: + +- `create` - Default is 15 minutes. +- `update` - Default is 15 minutes. +- `delete` - Default is 15 minutes. + +## Import + +Node can be imported using any of these accepted formats: + +``` +$ terraform import google_tpu_node.default projects/{{project}}/locations/{{zone}}/nodes/{{name}} +$ terraform import google_tpu_node.default {{project}}/{{zone}}/{{name}} +$ terraform import google_tpu_node.default {{name}} +``` + +-> If you're importing a resource with beta features, make sure to include `-provider=google-beta` +as an argument so that Terraform uses the correct provider to import your resource.