device_description

package
v0.4.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Sep 23, 2024 License: Apache-2.0 Imports: 5 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var File_xla_stream_executor_device_description_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type CudaComputeCapabilityProto 

type CudaComputeCapabilityProto struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	// contains filtered or unexported fields
}

func (*CudaComputeCapabilityProto) Descriptor deprecated 

func (*CudaComputeCapabilityProto) Descriptor() ([]byte, []int)

Deprecated: Use CudaComputeCapabilityProto.ProtoReflect.Descriptor instead.

func (*CudaComputeCapabilityProto) GetMajor 

func (x *CudaComputeCapabilityProto) GetMajor() int32

func (*CudaComputeCapabilityProto) GetMinor 

func (x *CudaComputeCapabilityProto) GetMinor() int32

func (*CudaComputeCapabilityProto) ProtoMessage 

func (*CudaComputeCapabilityProto) ProtoMessage()

func (*CudaComputeCapabilityProto) ProtoReflect 

func (x *CudaComputeCapabilityProto) ProtoReflect() protoreflect.Message

func (*CudaComputeCapabilityProto) Reset 

func (x *CudaComputeCapabilityProto) Reset()

func (*CudaComputeCapabilityProto) String 

func (x *CudaComputeCapabilityProto) String() string

type DnnVersionInfoProto 

type DnnVersionInfoProto struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	Patch int32 `protobuf:"varint,3,opt,name=patch,proto3" json:"patch,omitempty"`
	// contains filtered or unexported fields
}

func (*DnnVersionInfoProto) Descriptor deprecated 

func (*DnnVersionInfoProto) Descriptor() ([]byte, []int)

Deprecated: Use DnnVersionInfoProto.ProtoReflect.Descriptor instead.

func (*DnnVersionInfoProto) GetMajor 

func (x *DnnVersionInfoProto) GetMajor() int32

func (*DnnVersionInfoProto) GetMinor 

func (x *DnnVersionInfoProto) GetMinor() int32

func (*DnnVersionInfoProto) GetPatch 

func (x *DnnVersionInfoProto) GetPatch() int32

func (*DnnVersionInfoProto) ProtoMessage 

func (*DnnVersionInfoProto) ProtoMessage()

func (*DnnVersionInfoProto) ProtoReflect 

func (x *DnnVersionInfoProto) ProtoReflect() protoreflect.Message

func (*DnnVersionInfoProto) Reset 

func (x *DnnVersionInfoProto) Reset()

func (*DnnVersionInfoProto) String 

func (x *DnnVersionInfoProto) String() string

type GpuDeviceInfoProto 

type GpuDeviceInfoProto struct {
	ThreadsPerBlockLimit      int32   `` /* 126-byte string literal not displayed */
	ThreadsPerWarp            int32   `protobuf:"varint,2,opt,name=threads_per_warp,json=threadsPerWarp,proto3" json:"threads_per_warp,omitempty"`
	SharedMemoryPerBlock      int32   `` /* 126-byte string literal not displayed */
	SharedMemoryPerCore       int32   `protobuf:"varint,4,opt,name=shared_memory_per_core,json=sharedMemoryPerCore,proto3" json:"shared_memory_per_core,omitempty"`
	ThreadsPerCoreLimit       int32   `protobuf:"varint,5,opt,name=threads_per_core_limit,json=threadsPerCoreLimit,proto3" json:"threads_per_core_limit,omitempty"`
	CoreCount                 int32   `protobuf:"varint,6,opt,name=core_count,json=coreCount,proto3" json:"core_count,omitempty"`
	FpusPerCore               int64   `protobuf:"varint,7,opt,name=fpus_per_core,json=fpusPerCore,proto3" json:"fpus_per_core,omitempty"`
	BlockDimLimitX            int32   `protobuf:"varint,8,opt,name=block_dim_limit_x,json=blockDimLimitX,proto3" json:"block_dim_limit_x,omitempty"`
	BlockDimLimitY            int32   `protobuf:"varint,9,opt,name=block_dim_limit_y,json=blockDimLimitY,proto3" json:"block_dim_limit_y,omitempty"`
	BlockDimLimitZ            int32   `protobuf:"varint,10,opt,name=block_dim_limit_z,json=blockDimLimitZ,proto3" json:"block_dim_limit_z,omitempty"`
	MemoryBandwidth           int64   `protobuf:"varint,11,opt,name=memory_bandwidth,json=memoryBandwidth,proto3" json:"memory_bandwidth,omitempty"`
	L2CacheSize               int64   `protobuf:"varint,12,opt,name=l2_cache_size,json=l2CacheSize,proto3" json:"l2_cache_size,omitempty"`
	ClockRateGhz              float32 `protobuf:"fixed32,13,opt,name=clock_rate_ghz,json=clockRateGhz,proto3" json:"clock_rate_ghz,omitempty"`
	DeviceMemorySize          int64   `protobuf:"varint,14,opt,name=device_memory_size,json=deviceMemorySize,proto3" json:"device_memory_size,omitempty"`
	SharedMemoryPerBlockOptin int32   `` /* 144-byte string literal not displayed */
	// Types that are assignable to ComputeCapability:
	//
	//	*GpuDeviceInfoProto_CudaComputeCapability
	//	*GpuDeviceInfoProto_RocmComputeCapability
	ComputeCapability      isGpuDeviceInfoProto_ComputeCapability `protobuf_oneof:"compute_capability"`
	RegistersPerCoreLimit  int64                                  `` /* 130-byte string literal not displayed */
	RegistersPerBlockLimit int64                                  `` /* 133-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*GpuDeviceInfoProto) Descriptor deprecated 

func (*GpuDeviceInfoProto) Descriptor() ([]byte, []int)

Deprecated: Use GpuDeviceInfoProto.ProtoReflect.Descriptor instead.

func (*GpuDeviceInfoProto) GetBlockDimLimitX 

func (x *GpuDeviceInfoProto) GetBlockDimLimitX() int32

func (*GpuDeviceInfoProto) GetBlockDimLimitY 

func (x *GpuDeviceInfoProto) GetBlockDimLimitY() int32

func (*GpuDeviceInfoProto) GetBlockDimLimitZ 

func (x *GpuDeviceInfoProto) GetBlockDimLimitZ() int32

func (*GpuDeviceInfoProto) GetClockRateGhz 

func (x *GpuDeviceInfoProto) GetClockRateGhz() float32

func (*GpuDeviceInfoProto) GetComputeCapability 

func (m *GpuDeviceInfoProto) GetComputeCapability() isGpuDeviceInfoProto_ComputeCapability

func (*GpuDeviceInfoProto) GetCoreCount 

func (x *GpuDeviceInfoProto) GetCoreCount() int32

func (*GpuDeviceInfoProto) GetCudaComputeCapability 

func (x *GpuDeviceInfoProto) GetCudaComputeCapability() *CudaComputeCapabilityProto

func (*GpuDeviceInfoProto) GetDeviceMemorySize 

func (x *GpuDeviceInfoProto) GetDeviceMemorySize() int64

func (*GpuDeviceInfoProto) GetFpusPerCore 

func (x *GpuDeviceInfoProto) GetFpusPerCore() int64

func (*GpuDeviceInfoProto) GetL2CacheSize 

func (x *GpuDeviceInfoProto) GetL2CacheSize() int64

func (*GpuDeviceInfoProto) GetMemoryBandwidth 

func (x *GpuDeviceInfoProto) GetMemoryBandwidth() int64

func (*GpuDeviceInfoProto) GetRegistersPerBlockLimit 

func (x *GpuDeviceInfoProto) GetRegistersPerBlockLimit() int64

func (*GpuDeviceInfoProto) GetRegistersPerCoreLimit 

func (x *GpuDeviceInfoProto) GetRegistersPerCoreLimit() int64

func (*GpuDeviceInfoProto) GetRocmComputeCapability 

func (x *GpuDeviceInfoProto) GetRocmComputeCapability() *RocmComputeCapabilityProto

func (*GpuDeviceInfoProto) GetSharedMemoryPerBlock 

func (x *GpuDeviceInfoProto) GetSharedMemoryPerBlock() int32

func (*GpuDeviceInfoProto) GetSharedMemoryPerBlockOptin 

func (x *GpuDeviceInfoProto) GetSharedMemoryPerBlockOptin() int32

func (*GpuDeviceInfoProto) GetSharedMemoryPerCore 

func (x *GpuDeviceInfoProto) GetSharedMemoryPerCore() int32

func (*GpuDeviceInfoProto) GetThreadsPerBlockLimit 

func (x *GpuDeviceInfoProto) GetThreadsPerBlockLimit() int32

func (*GpuDeviceInfoProto) GetThreadsPerCoreLimit 

func (x *GpuDeviceInfoProto) GetThreadsPerCoreLimit() int32

func (*GpuDeviceInfoProto) GetThreadsPerWarp 

func (x *GpuDeviceInfoProto) GetThreadsPerWarp() int32

func (*GpuDeviceInfoProto) ProtoMessage 

func (*GpuDeviceInfoProto) ProtoMessage()

func (*GpuDeviceInfoProto) ProtoReflect 

func (x *GpuDeviceInfoProto) ProtoReflect() protoreflect.Message

func (*GpuDeviceInfoProto) Reset 

func (x *GpuDeviceInfoProto) Reset()

func (*GpuDeviceInfoProto) String 

func (x *GpuDeviceInfoProto) String() string

type GpuDeviceInfoProto_CudaComputeCapability 

type GpuDeviceInfoProto_CudaComputeCapability struct {
	CudaComputeCapability *CudaComputeCapabilityProto `protobuf:"bytes,16,opt,name=cuda_compute_capability,json=cudaComputeCapability,proto3,oneof"`
}

type GpuDeviceInfoProto_RocmComputeCapability 

type GpuDeviceInfoProto_RocmComputeCapability struct {
	RocmComputeCapability *RocmComputeCapabilityProto `protobuf:"bytes,17,opt,name=rocm_compute_capability,json=rocmComputeCapability,proto3,oneof"`
}

type GpuTargetConfigProto 

type GpuTargetConfigProto struct {
	GpuDeviceInfo  *GpuDeviceInfoProto  `protobuf:"bytes,1,opt,name=gpu_device_info,json=gpuDeviceInfo,proto3" json:"gpu_device_info,omitempty"`
	PlatformName   string               `protobuf:"bytes,4,opt,name=platform_name,json=platformName,proto3" json:"platform_name,omitempty"`
	DnnVersionInfo *DnnVersionInfoProto `protobuf:"bytes,5,opt,name=dnn_version_info,json=dnnVersionInfo,proto3" json:"dnn_version_info,omitempty"`
	// TODO(b/248362914): Autotuning results should be separate from
	// GpuTargetConfig because autotuning can be updated regularly separate from
	// the target.
	AutotuneResults      *autotune_results.AutotuneResults `protobuf:"bytes,6,opt,name=autotune_results,json=autotuneResults,proto3" json:"autotune_results,omitempty"`
	DeviceDescriptionStr string                            `protobuf:"bytes,7,opt,name=device_description_str,json=deviceDescriptionStr,proto3" json:"device_description_str,omitempty"`
	// contains filtered or unexported fields
}

func (*GpuTargetConfigProto) Descriptor deprecated 

func (*GpuTargetConfigProto) Descriptor() ([]byte, []int)

Deprecated: Use GpuTargetConfigProto.ProtoReflect.Descriptor instead.

func (*GpuTargetConfigProto) GetAutotuneResults 

func (x *GpuTargetConfigProto) GetAutotuneResults() *autotune_results.AutotuneResults

func (*GpuTargetConfigProto) GetDeviceDescriptionStr 

func (x *GpuTargetConfigProto) GetDeviceDescriptionStr() string

func (*GpuTargetConfigProto) GetDnnVersionInfo 

func (x *GpuTargetConfigProto) GetDnnVersionInfo() *DnnVersionInfoProto

func (*GpuTargetConfigProto) GetGpuDeviceInfo 

func (x *GpuTargetConfigProto) GetGpuDeviceInfo() *GpuDeviceInfoProto

func (*GpuTargetConfigProto) GetPlatformName 

func (x *GpuTargetConfigProto) GetPlatformName() string

func (*GpuTargetConfigProto) ProtoMessage 

func (*GpuTargetConfigProto) ProtoMessage()

func (*GpuTargetConfigProto) ProtoReflect 

func (x *GpuTargetConfigProto) ProtoReflect() protoreflect.Message

func (*GpuTargetConfigProto) Reset 

func (x *GpuTargetConfigProto) Reset()

func (*GpuTargetConfigProto) String 

func (x *GpuTargetConfigProto) String() string

type RocmComputeCapabilityProto 

type RocmComputeCapabilityProto struct {
	GcnArchName string `protobuf:"bytes,1,opt,name=gcn_arch_name,json=gcnArchName,proto3" json:"gcn_arch_name,omitempty"`
	// contains filtered or unexported fields
}

func (*RocmComputeCapabilityProto) Descriptor deprecated 

func (*RocmComputeCapabilityProto) Descriptor() ([]byte, []int)

Deprecated: Use RocmComputeCapabilityProto.ProtoReflect.Descriptor instead.

func (*RocmComputeCapabilityProto) GetGcnArchName 

func (x *RocmComputeCapabilityProto) GetGcnArchName() string

func (*RocmComputeCapabilityProto) ProtoMessage 

func (*RocmComputeCapabilityProto) ProtoMessage()

func (*RocmComputeCapabilityProto) ProtoReflect 

func (x *RocmComputeCapabilityProto) ProtoReflect() protoreflect.Message

func (*RocmComputeCapabilityProto) Reset 

func (x *RocmComputeCapabilityProto) Reset()

func (*RocmComputeCapabilityProto) String 

func (x *RocmComputeCapabilityProto) String() string

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.