// Copyright 2025 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// 	protoc-gen-go v1.36.11
// 	protoc        v6.33.2
// source: google/cloud/aiplatform/v1/machine_resources.proto

package aiplatformpb

import (
	reflect "reflect"
	sync "sync"
	unsafe "unsafe"

	_ "google.golang.org/genproto/googleapis/api/annotations"
	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
)

const (
	// Verify that this generated code is sufficiently up-to-date.
	_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
	// Verify that runtime/protoimpl is sufficiently up-to-date.
	_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)

// Specification of a single machine.
type MachineSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Immutable. The type of the machine.
	//
	// See the [list of machine types supported for
	// prediction](https://cloud.google.com/vertex-ai/docs/predictions/configure-compute#machine-types)
	//
	// See the [list of machine types supported for custom
	// training](https://cloud.google.com/vertex-ai/docs/training/configure-compute#machine-types).
	//
	// For [DeployedModel][google.cloud.aiplatform.v1.DeployedModel] this field is
	// optional, and the default value is `n1-standard-2`. For
	// [BatchPredictionJob][google.cloud.aiplatform.v1.BatchPredictionJob] or as
	// part of [WorkerPoolSpec][google.cloud.aiplatform.v1.WorkerPoolSpec] this
	// field is required.
	MachineType string `protobuf:"bytes,1,opt,name=machine_type,json=machineType,proto3" json:"machine_type,omitempty"`
	// Immutable. The type of accelerator(s) that may be attached to the machine
	// as per
	// [accelerator_count][google.cloud.aiplatform.v1.MachineSpec.accelerator_count].
	AcceleratorType AcceleratorType `protobuf:"varint,2,opt,name=accelerator_type,json=acceleratorType,proto3,enum=google.cloud.aiplatform.v1.AcceleratorType" json:"accelerator_type,omitempty"`
	// The number of accelerators to attach to the machine.
	AcceleratorCount int32 `protobuf:"varint,3,opt,name=accelerator_count,json=acceleratorCount,proto3" json:"accelerator_count,omitempty"`
	// Optional. Immutable. The Nvidia GPU partition size.
	//
	// When specified, the requested accelerators will be partitioned into
	// smaller GPU partitions. For example, if the request is for 8 units of
	// NVIDIA A100 GPUs, and gpu_partition_size="1g.10gb", the service will
	// create 8 * 7 = 56 partitioned MIG instances.
	//
	// The partition size must be a value supported by the requested accelerator.
	// Refer to
	// [Nvidia GPU
	// Partitioning](https://cloud.google.com/kubernetes-engine/docs/how-to/gpus-multi#multi-instance_gpu_partitions)
	// for the available partition sizes.
	//
	// If set, the accelerator_count should be set to 1.
	GpuPartitionSize string `protobuf:"bytes,7,opt,name=gpu_partition_size,json=gpuPartitionSize,proto3" json:"gpu_partition_size,omitempty"`
	// Immutable. The topology of the TPUs. Corresponds to the TPU topologies
	// available from GKE. (Example: tpu_topology: "2x2x1").
	TpuTopology string `protobuf:"bytes,4,opt,name=tpu_topology,json=tpuTopology,proto3" json:"tpu_topology,omitempty"`
	// Optional. Immutable. Configuration controlling how this resource pool
	// consumes reservation.
	ReservationAffinity *ReservationAffinity `protobuf:"bytes,5,opt,name=reservation_affinity,json=reservationAffinity,proto3" json:"reservation_affinity,omitempty"`
	unknownFields       protoimpl.UnknownFields
	sizeCache           protoimpl.SizeCache
}

func (x *MachineSpec) Reset() {
	*x = MachineSpec{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[0]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *MachineSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*MachineSpec) ProtoMessage() {}

func (x *MachineSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[0]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use MachineSpec.ProtoReflect.Descriptor instead.
func (*MachineSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{0}
}

func (x *MachineSpec) GetMachineType() string {
	if x != nil {
		return x.MachineType
	}
	return ""
}

func (x *MachineSpec) GetAcceleratorType() AcceleratorType {
	if x != nil {
		return x.AcceleratorType
	}
	return AcceleratorType_ACCELERATOR_TYPE_UNSPECIFIED
}

func (x *MachineSpec) GetAcceleratorCount() int32 {
	if x != nil {
		return x.AcceleratorCount
	}
	return 0
}

func (x *MachineSpec) GetGpuPartitionSize() string {
	if x != nil {
		return x.GpuPartitionSize
	}
	return ""
}

func (x *MachineSpec) GetTpuTopology() string {
	if x != nil {
		return x.TpuTopology
	}
	return ""
}

func (x *MachineSpec) GetReservationAffinity() *ReservationAffinity {
	if x != nil {
		return x.ReservationAffinity
	}
	return nil
}

// A description of resources that are dedicated to a DeployedModel or
// DeployedIndex, and that need a higher degree of manual configuration.
type DedicatedResources struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. Immutable. The specification of a single machine being used.
	MachineSpec *MachineSpec `protobuf:"bytes,1,opt,name=machine_spec,json=machineSpec,proto3" json:"machine_spec,omitempty"`
	// Required. Immutable. The minimum number of machine replicas that will be
	// always deployed on. This value must be greater than or equal to 1.
	//
	// If traffic increases, it may dynamically be deployed onto more replicas,
	// and as traffic decreases, some of these extra replicas may be freed.
	MinReplicaCount int32 `protobuf:"varint,2,opt,name=min_replica_count,json=minReplicaCount,proto3" json:"min_replica_count,omitempty"`
	// Immutable. The maximum number of replicas that may be deployed on when the
	// traffic against it increases. If the requested value is too large, the
	// deployment will error, but if deployment succeeds then the ability to scale
	// to that many replicas is guaranteed (barring service outages). If traffic
	// increases beyond what its replicas at maximum may handle, a portion of the
	// traffic will be dropped. If this value is not provided, will use
	// [min_replica_count][google.cloud.aiplatform.v1.DedicatedResources.min_replica_count]
	// as the default value.
	//
	// The value of this field impacts the charge against Vertex CPU and GPU
	// quotas. Specifically, you will be charged for (max_replica_count *
	// number of cores in the selected machine type) and (max_replica_count *
	// number of GPUs per replica in the selected machine type).
	MaxReplicaCount int32 `protobuf:"varint,3,opt,name=max_replica_count,json=maxReplicaCount,proto3" json:"max_replica_count,omitempty"`
	// Optional. Number of required available replicas for the deployment to
	// succeed. This field is only needed when partial deployment/mutation is
	// desired. If set, the deploy/mutate operation will succeed once
	// available_replica_count reaches required_replica_count, and the rest of
	// the replicas will be retried. If not set, the default
	// required_replica_count will be min_replica_count.
	RequiredReplicaCount int32 `protobuf:"varint,9,opt,name=required_replica_count,json=requiredReplicaCount,proto3" json:"required_replica_count,omitempty"`
	// Immutable. The metric specifications that overrides a resource
	// utilization metric (CPU utilization, accelerator's duty cycle, and so on)
	// target value (default to 60 if not set). At most one entry is allowed per
	// metric.
	//
	// If
	// [machine_spec.accelerator_count][google.cloud.aiplatform.v1.MachineSpec.accelerator_count]
	// is above 0, the autoscaling will be based on both CPU utilization and
	// accelerator's duty cycle metrics and scale up when either metrics exceeds
	// its target value while scale down if both metrics are under their target
	// value. The default target value is 60 for both metrics.
	//
	// If
	// [machine_spec.accelerator_count][google.cloud.aiplatform.v1.MachineSpec.accelerator_count]
	// is 0, the autoscaling will be based on CPU utilization metric only with
	// default target value 60 if not explicitly set.
	//
	// For example, in the case of Online Prediction, if you want to override
	// target CPU utilization to 80, you should set
	// [autoscaling_metric_specs.metric_name][google.cloud.aiplatform.v1.AutoscalingMetricSpec.metric_name]
	// to `aiplatform.googleapis.com/prediction/online/cpu/utilization` and
	// [autoscaling_metric_specs.target][google.cloud.aiplatform.v1.AutoscalingMetricSpec.target]
	// to `80`.
	AutoscalingMetricSpecs []*AutoscalingMetricSpec `protobuf:"bytes,4,rep,name=autoscaling_metric_specs,json=autoscalingMetricSpecs,proto3" json:"autoscaling_metric_specs,omitempty"`
	// Optional. If true, schedule the deployment workload on [spot
	// VMs](https://cloud.google.com/kubernetes-engine/docs/concepts/spot-vms).
	Spot          bool `protobuf:"varint,5,opt,name=spot,proto3" json:"spot,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *DedicatedResources) Reset() {
	*x = DedicatedResources{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[1]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *DedicatedResources) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*DedicatedResources) ProtoMessage() {}

func (x *DedicatedResources) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[1]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use DedicatedResources.ProtoReflect.Descriptor instead.
func (*DedicatedResources) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{1}
}

func (x *DedicatedResources) GetMachineSpec() *MachineSpec {
	if x != nil {
		return x.MachineSpec
	}
	return nil
}

func (x *DedicatedResources) GetMinReplicaCount() int32 {
	if x != nil {
		return x.MinReplicaCount
	}
	return 0
}

func (x *DedicatedResources) GetMaxReplicaCount() int32 {
	if x != nil {
		return x.MaxReplicaCount
	}
	return 0
}

func (x *DedicatedResources) GetRequiredReplicaCount() int32 {
	if x != nil {
		return x.RequiredReplicaCount
	}
	return 0
}

func (x *DedicatedResources) GetAutoscalingMetricSpecs() []*AutoscalingMetricSpec {
	if x != nil {
		return x.AutoscalingMetricSpecs
	}
	return nil
}

func (x *DedicatedResources) GetSpot() bool {
	if x != nil {
		return x.Spot
	}
	return false
}

// A description of resources that to large degree are decided by Vertex AI,
// and require only a modest additional configuration.
// Each Model supporting these resources documents its specific guidelines.
type AutomaticResources struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Immutable. The minimum number of replicas that will be always deployed on.
	// If traffic against it increases, it may dynamically be deployed onto more
	// replicas up to
	// [max_replica_count][google.cloud.aiplatform.v1.AutomaticResources.max_replica_count],
	// and as traffic decreases, some of these extra replicas may be freed. If the
	// requested value is too large, the deployment will error.
	MinReplicaCount int32 `protobuf:"varint,1,opt,name=min_replica_count,json=minReplicaCount,proto3" json:"min_replica_count,omitempty"`
	// Immutable. The maximum number of replicas that may be deployed on when the
	// traffic against it increases. If the requested value is too large, the
	// deployment will error, but if deployment succeeds then the ability to scale
	// to that many replicas is guaranteed (barring service outages). If traffic
	// increases beyond what its replicas at maximum may handle, a portion of the
	// traffic will be dropped. If this value is not provided, a no upper bound
	// for scaling under heavy traffic will be assume, though Vertex AI may be
	// unable to scale beyond certain replica number.
	MaxReplicaCount int32 `protobuf:"varint,2,opt,name=max_replica_count,json=maxReplicaCount,proto3" json:"max_replica_count,omitempty"`
	unknownFields   protoimpl.UnknownFields
	sizeCache       protoimpl.SizeCache
}

func (x *AutomaticResources) Reset() {
	*x = AutomaticResources{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[2]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *AutomaticResources) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*AutomaticResources) ProtoMessage() {}

func (x *AutomaticResources) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[2]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use AutomaticResources.ProtoReflect.Descriptor instead.
func (*AutomaticResources) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{2}
}

func (x *AutomaticResources) GetMinReplicaCount() int32 {
	if x != nil {
		return x.MinReplicaCount
	}
	return 0
}

func (x *AutomaticResources) GetMaxReplicaCount() int32 {
	if x != nil {
		return x.MaxReplicaCount
	}
	return 0
}

// A description of resources that are used for performing batch operations, are
// dedicated to a Model, and need manual configuration.
type BatchDedicatedResources struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. Immutable. The specification of a single machine.
	MachineSpec *MachineSpec `protobuf:"bytes,1,opt,name=machine_spec,json=machineSpec,proto3" json:"machine_spec,omitempty"`
	// Immutable. The number of machine replicas used at the start of the batch
	// operation. If not set, Vertex AI decides starting number, not greater than
	// [max_replica_count][google.cloud.aiplatform.v1.BatchDedicatedResources.max_replica_count]
	StartingReplicaCount int32 `protobuf:"varint,2,opt,name=starting_replica_count,json=startingReplicaCount,proto3" json:"starting_replica_count,omitempty"`
	// Immutable. The maximum number of machine replicas the batch operation may
	// be scaled to. The default value is 10.
	MaxReplicaCount int32 `protobuf:"varint,3,opt,name=max_replica_count,json=maxReplicaCount,proto3" json:"max_replica_count,omitempty"`
	unknownFields   protoimpl.UnknownFields
	sizeCache       protoimpl.SizeCache
}

func (x *BatchDedicatedResources) Reset() {
	*x = BatchDedicatedResources{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[3]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *BatchDedicatedResources) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*BatchDedicatedResources) ProtoMessage() {}

func (x *BatchDedicatedResources) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[3]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use BatchDedicatedResources.ProtoReflect.Descriptor instead.
func (*BatchDedicatedResources) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{3}
}

func (x *BatchDedicatedResources) GetMachineSpec() *MachineSpec {
	if x != nil {
		return x.MachineSpec
	}
	return nil
}

func (x *BatchDedicatedResources) GetStartingReplicaCount() int32 {
	if x != nil {
		return x.StartingReplicaCount
	}
	return 0
}

func (x *BatchDedicatedResources) GetMaxReplicaCount() int32 {
	if x != nil {
		return x.MaxReplicaCount
	}
	return 0
}

// Statistics information about resource consumption.
type ResourcesConsumed struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Output only. The number of replica hours used. Note that many replicas may
	// run in parallel, and additionally any given work may be queued for some
	// time. Therefore this value is not strictly related to wall time.
	ReplicaHours  float64 `protobuf:"fixed64,1,opt,name=replica_hours,json=replicaHours,proto3" json:"replica_hours,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *ResourcesConsumed) Reset() {
	*x = ResourcesConsumed{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[4]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *ResourcesConsumed) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*ResourcesConsumed) ProtoMessage() {}

func (x *ResourcesConsumed) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[4]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use ResourcesConsumed.ProtoReflect.Descriptor instead.
func (*ResourcesConsumed) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{4}
}

func (x *ResourcesConsumed) GetReplicaHours() float64 {
	if x != nil {
		return x.ReplicaHours
	}
	return 0
}

// Represents the spec of disk options.
type DiskSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Type of the boot disk. For non-A3U machines, the default value is
	// "pd-ssd", for A3U machines, the default value is "hyperdisk-balanced".
	// Valid values: "pd-ssd" (Persistent Disk Solid State Drive),
	// "pd-standard" (Persistent Disk Hard Disk Drive) or "hyperdisk-balanced".
	BootDiskType string `protobuf:"bytes,1,opt,name=boot_disk_type,json=bootDiskType,proto3" json:"boot_disk_type,omitempty"`
	// Size in GB of the boot disk (default is 100GB).
	BootDiskSizeGb int32 `protobuf:"varint,2,opt,name=boot_disk_size_gb,json=bootDiskSizeGb,proto3" json:"boot_disk_size_gb,omitempty"`
	unknownFields  protoimpl.UnknownFields
	sizeCache      protoimpl.SizeCache
}

func (x *DiskSpec) Reset() {
	*x = DiskSpec{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[5]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *DiskSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*DiskSpec) ProtoMessage() {}

func (x *DiskSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[5]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use DiskSpec.ProtoReflect.Descriptor instead.
func (*DiskSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{5}
}

func (x *DiskSpec) GetBootDiskType() string {
	if x != nil {
		return x.BootDiskType
	}
	return ""
}

func (x *DiskSpec) GetBootDiskSizeGb() int32 {
	if x != nil {
		return x.BootDiskSizeGb
	}
	return 0
}

// Represents the spec of [persistent
// disk][https://cloud.google.com/compute/docs/disks/persistent-disks] options.
type PersistentDiskSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Type of the disk (default is "pd-standard").
	// Valid values: "pd-ssd" (Persistent Disk Solid State Drive)
	// "pd-standard" (Persistent Disk Hard Disk Drive)
	// "pd-balanced" (Balanced Persistent Disk)
	// "pd-extreme" (Extreme Persistent Disk)
	DiskType string `protobuf:"bytes,1,opt,name=disk_type,json=diskType,proto3" json:"disk_type,omitempty"`
	// Size in GB of the disk (default is 100GB).
	DiskSizeGb    int64 `protobuf:"varint,2,opt,name=disk_size_gb,json=diskSizeGb,proto3" json:"disk_size_gb,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *PersistentDiskSpec) Reset() {
	*x = PersistentDiskSpec{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[6]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *PersistentDiskSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*PersistentDiskSpec) ProtoMessage() {}

func (x *PersistentDiskSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[6]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use PersistentDiskSpec.ProtoReflect.Descriptor instead.
func (*PersistentDiskSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{6}
}

func (x *PersistentDiskSpec) GetDiskType() string {
	if x != nil {
		return x.DiskType
	}
	return ""
}

func (x *PersistentDiskSpec) GetDiskSizeGb() int64 {
	if x != nil {
		return x.DiskSizeGb
	}
	return 0
}

// Represents a mount configuration for Network File System (NFS) to mount.
type NfsMount struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. IP address of the NFS server.
	Server string `protobuf:"bytes,1,opt,name=server,proto3" json:"server,omitempty"`
	// Required. Source path exported from NFS server.
	// Has to start with '/', and combined with the ip address, it indicates
	// the source mount path in the form of `server:path`
	Path string `protobuf:"bytes,2,opt,name=path,proto3" json:"path,omitempty"`
	// Required. Destination mount path. The NFS will be mounted for the user
	// under /mnt/nfs/<mount_point>
	MountPoint    string `protobuf:"bytes,3,opt,name=mount_point,json=mountPoint,proto3" json:"mount_point,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NfsMount) Reset() {
	*x = NfsMount{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[7]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NfsMount) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NfsMount) ProtoMessage() {}

func (x *NfsMount) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[7]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NfsMount.ProtoReflect.Descriptor instead.
func (*NfsMount) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{7}
}

func (x *NfsMount) GetServer() string {
	if x != nil {
		return x.Server
	}
	return ""
}

func (x *NfsMount) GetPath() string {
	if x != nil {
		return x.Path
	}
	return ""
}

func (x *NfsMount) GetMountPoint() string {
	if x != nil {
		return x.MountPoint
	}
	return ""
}

// Represents a mount configuration for Lustre file system.
type LustreMount struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. IP address of the Lustre instance.
	InstanceIp string `protobuf:"bytes,1,opt,name=instance_ip,json=instanceIp,proto3" json:"instance_ip,omitempty"`
	// Required. The unique identifier of the Lustre volume.
	VolumeHandle string `protobuf:"bytes,2,opt,name=volume_handle,json=volumeHandle,proto3" json:"volume_handle,omitempty"`
	// Required. The name of the Lustre filesystem.
	Filesystem string `protobuf:"bytes,3,opt,name=filesystem,proto3" json:"filesystem,omitempty"`
	// Required. Destination mount path. The Lustre file system will be mounted
	// for the user under /mnt/lustre/<mount_point>
	MountPoint    string `protobuf:"bytes,4,opt,name=mount_point,json=mountPoint,proto3" json:"mount_point,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *LustreMount) Reset() {
	*x = LustreMount{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[8]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *LustreMount) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*LustreMount) ProtoMessage() {}

func (x *LustreMount) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[8]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use LustreMount.ProtoReflect.Descriptor instead.
func (*LustreMount) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{8}
}

func (x *LustreMount) GetInstanceIp() string {
	if x != nil {
		return x.InstanceIp
	}
	return ""
}

func (x *LustreMount) GetVolumeHandle() string {
	if x != nil {
		return x.VolumeHandle
	}
	return ""
}

func (x *LustreMount) GetFilesystem() string {
	if x != nil {
		return x.Filesystem
	}
	return ""
}

func (x *LustreMount) GetMountPoint() string {
	if x != nil {
		return x.MountPoint
	}
	return ""
}

// The metric specification that defines the target resource utilization
// (CPU utilization, accelerator's duty cycle, and so on) for calculating the
// desired replica count.
type AutoscalingMetricSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. The resource metric name.
	// Supported metrics:
	//
	// * For Online Prediction:
	// * `aiplatform.googleapis.com/prediction/online/accelerator/duty_cycle`
	// * `aiplatform.googleapis.com/prediction/online/cpu/utilization`
	// * `aiplatform.googleapis.com/prediction/online/request_count`
	MetricName string `protobuf:"bytes,1,opt,name=metric_name,json=metricName,proto3" json:"metric_name,omitempty"`
	// The target resource utilization in percentage (1% - 100%) for the given
	// metric; once the real usage deviates from the target by a certain
	// percentage, the machine replicas change. The default value is 60
	// (representing 60%) if not provided.
	Target        int32 `protobuf:"varint,2,opt,name=target,proto3" json:"target,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *AutoscalingMetricSpec) Reset() {
	*x = AutoscalingMetricSpec{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[9]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *AutoscalingMetricSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*AutoscalingMetricSpec) ProtoMessage() {}

func (x *AutoscalingMetricSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[9]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use AutoscalingMetricSpec.ProtoReflect.Descriptor instead.
func (*AutoscalingMetricSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{9}
}

func (x *AutoscalingMetricSpec) GetMetricName() string {
	if x != nil {
		return x.MetricName
	}
	return ""
}

func (x *AutoscalingMetricSpec) GetTarget() int32 {
	if x != nil {
		return x.Target
	}
	return 0
}

// A set of Shielded Instance options.
// See [Images using supported Shielded VM
// features](https://cloud.google.com/compute/docs/instances/modifying-shielded-vm).
type ShieldedVmConfig struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Defines whether the instance has [Secure
	// Boot](https://cloud.google.com/compute/shielded-vm/docs/shielded-vm#secure-boot)
	// enabled.
	//
	// Secure Boot helps ensure that the system only runs authentic software by
	// verifying the digital signature of all boot components, and halting the
	// boot process if signature verification fails.
	EnableSecureBoot bool `protobuf:"varint,1,opt,name=enable_secure_boot,json=enableSecureBoot,proto3" json:"enable_secure_boot,omitempty"`
	unknownFields    protoimpl.UnknownFields
	sizeCache        protoimpl.SizeCache
}

func (x *ShieldedVmConfig) Reset() {
	*x = ShieldedVmConfig{}
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[10]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *ShieldedVmConfig) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*ShieldedVmConfig) ProtoMessage() {}

func (x *ShieldedVmConfig) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes[10]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use ShieldedVmConfig.ProtoReflect.Descriptor instead.
func (*ShieldedVmConfig) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP(), []int{10}
}

func (x *ShieldedVmConfig) GetEnableSecureBoot() bool {
	if x != nil {
		return x.EnableSecureBoot
	}
	return false
}

var File_google_cloud_aiplatform_v1_machine_resources_proto protoreflect.FileDescriptor

const file_google_cloud_aiplatform_v1_machine_resources_proto_rawDesc = "" +
	"\n" +
	"2google/cloud/aiplatform/v1/machine_resources.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a1google/cloud/aiplatform/v1/accelerator_type.proto\x1a5google/cloud/aiplatform/v1/reservation_affinity.proto\"\x89\x03\n" +
	"\vMachineSpec\x12&\n" +
	"\fmachine_type\x18\x01 \x01(\tB\x03\xe0A\x05R\vmachineType\x12[\n" +
	"\x10accelerator_type\x18\x02 \x01(\x0e2+.google.cloud.aiplatform.v1.AcceleratorTypeB\x03\xe0A\x05R\x0facceleratorType\x12+\n" +
	"\x11accelerator_count\x18\x03 \x01(\x05R\x10acceleratorCount\x124\n" +
	"\x12gpu_partition_size\x18\a \x01(\tB\x06\xe0A\x05\xe0A\x01R\x10gpuPartitionSize\x12&\n" +
	"\ftpu_topology\x18\x04 \x01(\tB\x03\xe0A\x05R\vtpuTopology\x12j\n" +
	"\x14reservation_affinity\x18\x05 \x01(\v2/.google.cloud.aiplatform.v1.ReservationAffinityB\x06\xe0A\x05\xe0A\x01R\x13reservationAffinity\"\x93\x03\n" +
	"\x12DedicatedResources\x12R\n" +
	"\fmachine_spec\x18\x01 \x01(\v2'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0A\x02\xe0A\x05R\vmachineSpec\x122\n" +
	"\x11min_replica_count\x18\x02 \x01(\x05B\x06\xe0A\x02\xe0A\x05R\x0fminReplicaCount\x12/\n" +
	"\x11max_replica_count\x18\x03 \x01(\x05B\x03\xe0A\x05R\x0fmaxReplicaCount\x129\n" +
	"\x16required_replica_count\x18\t \x01(\x05B\x03\xe0A\x01R\x14requiredReplicaCount\x12p\n" +
	"\x18autoscaling_metric_specs\x18\x04 \x03(\v21.google.cloud.aiplatform.v1.AutoscalingMetricSpecB\x03\xe0A\x05R\x16autoscalingMetricSpecs\x12\x17\n" +
	"\x04spot\x18\x05 \x01(\bB\x03\xe0A\x01R\x04spot\"v\n" +
	"\x12AutomaticResources\x12/\n" +
	"\x11min_replica_count\x18\x01 \x01(\x05B\x03\xe0A\x05R\x0fminReplicaCount\x12/\n" +
	"\x11max_replica_count\x18\x02 \x01(\x05B\x03\xe0A\x05R\x0fmaxReplicaCount\"\xd9\x01\n" +
	"\x17BatchDedicatedResources\x12R\n" +
	"\fmachine_spec\x18\x01 \x01(\v2'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0A\x02\xe0A\x05R\vmachineSpec\x129\n" +
	"\x16starting_replica_count\x18\x02 \x01(\x05B\x03\xe0A\x05R\x14startingReplicaCount\x12/\n" +
	"\x11max_replica_count\x18\x03 \x01(\x05B\x03\xe0A\x05R\x0fmaxReplicaCount\"=\n" +
	"\x11ResourcesConsumed\x12(\n" +
	"\rreplica_hours\x18\x01 \x01(\x01B\x03\xe0A\x03R\freplicaHours\"[\n" +
	"\bDiskSpec\x12$\n" +
	"\x0eboot_disk_type\x18\x01 \x01(\tR\fbootDiskType\x12)\n" +
	"\x11boot_disk_size_gb\x18\x02 \x01(\x05R\x0ebootDiskSizeGb\"S\n" +
	"\x12PersistentDiskSpec\x12\x1b\n" +
	"\tdisk_type\x18\x01 \x01(\tR\bdiskType\x12 \n" +
	"\fdisk_size_gb\x18\x02 \x01(\x03R\n" +
	"diskSizeGb\"f\n" +
	"\bNfsMount\x12\x1b\n" +
	"\x06server\x18\x01 \x01(\tB\x03\xe0A\x02R\x06server\x12\x17\n" +
	"\x04path\x18\x02 \x01(\tB\x03\xe0A\x02R\x04path\x12$\n" +
	"\vmount_point\x18\x03 \x01(\tB\x03\xe0A\x02R\n" +
	"mountPoint\"\xa8\x01\n" +
	"\vLustreMount\x12$\n" +
	"\vinstance_ip\x18\x01 \x01(\tB\x03\xe0A\x02R\n" +
	"instanceIp\x12(\n" +
	"\rvolume_handle\x18\x02 \x01(\tB\x03\xe0A\x02R\fvolumeHandle\x12#\n" +
	"\n" +
	"filesystem\x18\x03 \x01(\tB\x03\xe0A\x02R\n" +
	"filesystem\x12$\n" +
	"\vmount_point\x18\x04 \x01(\tB\x03\xe0A\x02R\n" +
	"mountPoint\"U\n" +
	"\x15AutoscalingMetricSpec\x12$\n" +
	"\vmetric_name\x18\x01 \x01(\tB\x03\xe0A\x02R\n" +
	"metricName\x12\x16\n" +
	"\x06target\x18\x02 \x01(\x05R\x06target\"@\n" +
	"\x10ShieldedVmConfig\x12,\n" +
	"\x12enable_secure_boot\x18\x01 \x01(\bR\x10enableSecureBootB\xd3\x01\n" +
	"\x1ecom.google.cloud.aiplatform.v1B\x15MachineResourcesProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"

var (
	file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescOnce sync.Once
	file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescData []byte
)

func file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescGZIP() []byte {
	file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescOnce.Do(func() {
		file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_google_cloud_aiplatform_v1_machine_resources_proto_rawDesc), len(file_google_cloud_aiplatform_v1_machine_resources_proto_rawDesc)))
	})
	return file_google_cloud_aiplatform_v1_machine_resources_proto_rawDescData
}

var file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes = make([]protoimpl.MessageInfo, 11)
var file_google_cloud_aiplatform_v1_machine_resources_proto_goTypes = []any{
	(*MachineSpec)(nil),             // 0: google.cloud.aiplatform.v1.MachineSpec
	(*DedicatedResources)(nil),      // 1: google.cloud.aiplatform.v1.DedicatedResources
	(*AutomaticResources)(nil),      // 2: google.cloud.aiplatform.v1.AutomaticResources
	(*BatchDedicatedResources)(nil), // 3: google.cloud.aiplatform.v1.BatchDedicatedResources
	(*ResourcesConsumed)(nil),       // 4: google.cloud.aiplatform.v1.ResourcesConsumed
	(*DiskSpec)(nil),                // 5: google.cloud.aiplatform.v1.DiskSpec
	(*PersistentDiskSpec)(nil),      // 6: google.cloud.aiplatform.v1.PersistentDiskSpec
	(*NfsMount)(nil),                // 7: google.cloud.aiplatform.v1.NfsMount
	(*LustreMount)(nil),             // 8: google.cloud.aiplatform.v1.LustreMount
	(*AutoscalingMetricSpec)(nil),   // 9: google.cloud.aiplatform.v1.AutoscalingMetricSpec
	(*ShieldedVmConfig)(nil),        // 10: google.cloud.aiplatform.v1.ShieldedVmConfig
	(AcceleratorType)(0),            // 11: google.cloud.aiplatform.v1.AcceleratorType
	(*ReservationAffinity)(nil),     // 12: google.cloud.aiplatform.v1.ReservationAffinity
}
var file_google_cloud_aiplatform_v1_machine_resources_proto_depIdxs = []int32{
	11, // 0: google.cloud.aiplatform.v1.MachineSpec.accelerator_type:type_name -> google.cloud.aiplatform.v1.AcceleratorType
	12, // 1: google.cloud.aiplatform.v1.MachineSpec.reservation_affinity:type_name -> google.cloud.aiplatform.v1.ReservationAffinity
	0,  // 2: google.cloud.aiplatform.v1.DedicatedResources.machine_spec:type_name -> google.cloud.aiplatform.v1.MachineSpec
	9,  // 3: google.cloud.aiplatform.v1.DedicatedResources.autoscaling_metric_specs:type_name -> google.cloud.aiplatform.v1.AutoscalingMetricSpec
	0,  // 4: google.cloud.aiplatform.v1.BatchDedicatedResources.machine_spec:type_name -> google.cloud.aiplatform.v1.MachineSpec
	5,  // [5:5] is the sub-list for method output_type
	5,  // [5:5] is the sub-list for method input_type
	5,  // [5:5] is the sub-list for extension type_name
	5,  // [5:5] is the sub-list for extension extendee
	0,  // [0:5] is the sub-list for field type_name
}

func init() { file_google_cloud_aiplatform_v1_machine_resources_proto_init() }
func file_google_cloud_aiplatform_v1_machine_resources_proto_init() {
	if File_google_cloud_aiplatform_v1_machine_resources_proto != nil {
		return
	}
	file_google_cloud_aiplatform_v1_accelerator_type_proto_init()
	file_google_cloud_aiplatform_v1_reservation_affinity_proto_init()
	type x struct{}
	out := protoimpl.TypeBuilder{
		File: protoimpl.DescBuilder{
			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
			RawDescriptor: unsafe.Slice(unsafe.StringData(file_google_cloud_aiplatform_v1_machine_resources_proto_rawDesc), len(file_google_cloud_aiplatform_v1_machine_resources_proto_rawDesc)),
			NumEnums:      0,
			NumMessages:   11,
			NumExtensions: 0,
			NumServices:   0,
		},
		GoTypes:           file_google_cloud_aiplatform_v1_machine_resources_proto_goTypes,
		DependencyIndexes: file_google_cloud_aiplatform_v1_machine_resources_proto_depIdxs,
		MessageInfos:      file_google_cloud_aiplatform_v1_machine_resources_proto_msgTypes,
	}.Build()
	File_google_cloud_aiplatform_v1_machine_resources_proto = out.File
	file_google_cloud_aiplatform_v1_machine_resources_proto_goTypes = nil
	file_google_cloud_aiplatform_v1_machine_resources_proto_depIdxs = nil
}
