gpu

package
v0.9.219 Latest Latest
Warning

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

 Go to latest Published: Aug 14, 2024 License: AGPL-3.0 Imports: 8 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var CedanaGPU_ServiceDesc = grpc.ServiceDesc{
	ServiceName: "cedanagpu.CedanaGPU",
	HandlerType: (*CedanaGPUServer)(nil),
	Methods: []grpc.MethodDesc{
		{
			MethodName: "Checkpoint",
			Handler:    _CedanaGPU_Checkpoint_Handler,
		},
		{
			MethodName: "Restore",
			Handler:    _CedanaGPU_Restore_Handler,
		},
		{
			MethodName: "StartupPoll",
			Handler:    _CedanaGPU_StartupPoll_Handler,
		},
		{
			MethodName: "HealthCheck",
			Handler:    _CedanaGPU_HealthCheck_Handler,
		},
	},
	Streams:  []grpc.StreamDesc{},
	Metadata: "gpu.proto",
}

CedanaGPU_ServiceDesc is the grpc.ServiceDesc for CedanaGPU service. It's only intended for direct use with grpc.RegisterService, and not to be introspected or modified (even as a copy)

View Source 
var File_gpu_proto protoreflect.FileDescriptor

Functions

func RegisterCedanaGPUServer 

func RegisterCedanaGPUServer(s grpc.ServiceRegistrar, srv CedanaGPUServer)

Types

type AvailableCUDAAPIs added in v0.9.211 

type AvailableCUDAAPIs struct {
	CuDNNVersion   int32 `protobuf:"varint,1,opt,name=cuDNNVersion,proto3" json:"cuDNNVersion,omitempty"`
	CuBLASVersion  int32 `protobuf:"varint,2,opt,name=cuBLASVersion,proto3" json:"cuBLASVersion,omitempty"`
	NcclVersion    int32 `protobuf:"varint,3,opt,name=ncclVersion,proto3" json:"ncclVersion,omitempty"`
	DriverVersion  int32 `protobuf:"varint,4,opt,name=driverVersion,proto3" json:"driverVersion,omitempty"`
	RuntimeVersion int32 `protobuf:"varint,5,opt,name=runtimeVersion,proto3" json:"runtimeVersion,omitempty"`
	// contains filtered or unexported fields
}

func (*AvailableCUDAAPIs) Descriptor deprecated added in v0.9.211 

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

Deprecated: Use AvailableCUDAAPIs.ProtoReflect.Descriptor instead.

func (*AvailableCUDAAPIs) GetCuBLASVersion added in v0.9.211 

func (x *AvailableCUDAAPIs) GetCuBLASVersion() int32

func (*AvailableCUDAAPIs) GetCuDNNVersion added in v0.9.211 

func (x *AvailableCUDAAPIs) GetCuDNNVersion() int32

func (*AvailableCUDAAPIs) GetDriverVersion added in v0.9.211 

func (x *AvailableCUDAAPIs) GetDriverVersion() int32

func (*AvailableCUDAAPIs) GetNcclVersion added in v0.9.211 

func (x *AvailableCUDAAPIs) GetNcclVersion() int32

func (*AvailableCUDAAPIs) GetRuntimeVersion added in v0.9.211 

func (x *AvailableCUDAAPIs) GetRuntimeVersion() int32

func (*AvailableCUDAAPIs) ProtoMessage added in v0.9.211 

func (*AvailableCUDAAPIs) ProtoMessage()

func (*AvailableCUDAAPIs) ProtoReflect added in v0.9.211 

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

func (*AvailableCUDAAPIs) Reset added in v0.9.211 

func (x *AvailableCUDAAPIs) Reset()

func (*AvailableCUDAAPIs) String added in v0.9.211 

func (x *AvailableCUDAAPIs) String() string

type CedanaGPUClient 

type CedanaGPUClient interface {
	Checkpoint(ctx context.Context, in *CheckpointRequest, opts ...grpc.CallOption) (*CheckpointResponse, error)
	Restore(ctx context.Context, in *RestoreRequest, opts ...grpc.CallOption) (*RestoreResponse, error)
	StartupPoll(ctx context.Context, in *StartupPollRequest, opts ...grpc.CallOption) (*StartupPollResponse, error)
	HealthCheck(ctx context.Context, in *HealthCheckRequest, opts ...grpc.CallOption) (*HealthCheckResponse, error)
}

CedanaGPUClient is the client API for CedanaGPU service.

For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.

func NewCedanaGPUClient 

func NewCedanaGPUClient(cc grpc.ClientConnInterface) CedanaGPUClient

type CedanaGPUServer 

type CedanaGPUServer interface {
	Checkpoint(context.Context, *CheckpointRequest) (*CheckpointResponse, error)
	Restore(context.Context, *RestoreRequest) (*RestoreResponse, error)
	StartupPoll(context.Context, *StartupPollRequest) (*StartupPollResponse, error)
	HealthCheck(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error)
	// contains filtered or unexported methods
}

CedanaGPUServer is the server API for CedanaGPU service. All implementations must embed UnimplementedCedanaGPUServer for forward compatibility

type CheckpointRequest 

type CheckpointRequest struct {
	Directory string `protobuf:"bytes,1,opt,name=directory,proto3" json:"directory,omitempty"`
	// contains filtered or unexported fields
}

func (*CheckpointRequest) Descriptor deprecated 

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

Deprecated: Use CheckpointRequest.ProtoReflect.Descriptor instead.

func (*CheckpointRequest) GetDirectory added in v0.9.5 

func (x *CheckpointRequest) GetDirectory() string

func (*CheckpointRequest) ProtoMessage 

func (*CheckpointRequest) ProtoMessage()

func (*CheckpointRequest) ProtoReflect 

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

func (*CheckpointRequest) Reset 

func (x *CheckpointRequest) Reset()

func (*CheckpointRequest) String 

func (x *CheckpointRequest) String() string

type CheckpointResponse 

type CheckpointResponse struct {
	Success  bool   `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`
	MemPath  string `protobuf:"bytes,2,opt,name=memPath,proto3" json:"memPath,omitempty"`
	CkptPath string `protobuf:"bytes,3,opt,name=ckptPath,proto3" json:"ckptPath,omitempty"`
	// contains filtered or unexported fields
}

func (*CheckpointResponse) Descriptor deprecated 

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

Deprecated: Use CheckpointResponse.ProtoReflect.Descriptor instead.

func (*CheckpointResponse) GetCkptPath added in v0.9.98 

func (x *CheckpointResponse) GetCkptPath() string

func (*CheckpointResponse) GetMemPath added in v0.9.98 

func (x *CheckpointResponse) GetMemPath() string

func (*CheckpointResponse) GetSuccess 

func (x *CheckpointResponse) GetSuccess() bool

func (*CheckpointResponse) ProtoMessage 

func (*CheckpointResponse) ProtoMessage()

func (*CheckpointResponse) ProtoReflect 

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

func (*CheckpointResponse) Reset 

func (x *CheckpointResponse) Reset()

func (*CheckpointResponse) String 

func (x *CheckpointResponse) String() string

type GPURestoreStats added in v0.9.216 

type GPURestoreStats struct {
	CopyMemTime     int64 `protobuf:"varint,1,opt,name=copyMemTime,proto3" json:"copyMemTime,omitempty"`
	ReplayCallsTime int64 `protobuf:"varint,2,opt,name=replayCallsTime,proto3" json:"replayCallsTime,omitempty"`
	// contains filtered or unexported fields
}

func (*GPURestoreStats) Descriptor deprecated added in v0.9.216 

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

Deprecated: Use GPURestoreStats.ProtoReflect.Descriptor instead.

func (*GPURestoreStats) GetCopyMemTime added in v0.9.216 

func (x *GPURestoreStats) GetCopyMemTime() int64

func (*GPURestoreStats) GetReplayCallsTime added in v0.9.216 

func (x *GPURestoreStats) GetReplayCallsTime() int64

func (*GPURestoreStats) ProtoMessage added in v0.9.216 

func (*GPURestoreStats) ProtoMessage()

func (*GPURestoreStats) ProtoReflect added in v0.9.216 

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

func (*GPURestoreStats) Reset added in v0.9.216 

func (x *GPURestoreStats) Reset()

func (*GPURestoreStats) String added in v0.9.216 

func (x *GPURestoreStats) String() string

type HealthCheckRequest added in v0.9.211 

type HealthCheckRequest struct {
	// contains filtered or unexported fields
}

func (*HealthCheckRequest) Descriptor deprecated added in v0.9.211 

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

Deprecated: Use HealthCheckRequest.ProtoReflect.Descriptor instead.

func (*HealthCheckRequest) ProtoMessage added in v0.9.211 

func (*HealthCheckRequest) ProtoMessage()

func (*HealthCheckRequest) ProtoReflect added in v0.9.211 

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

func (*HealthCheckRequest) Reset added in v0.9.211 

func (x *HealthCheckRequest) Reset()

func (*HealthCheckRequest) String added in v0.9.211 

func (x *HealthCheckRequest) String() string

type HealthCheckResponse added in v0.9.211 

type HealthCheckResponse struct {
	Success           bool               `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`
	Version           string             `protobuf:"bytes,2,opt,name=version,proto3" json:"version,omitempty"`
	DeviceCount       int32              `protobuf:"varint,3,opt,name=deviceCount,proto3" json:"deviceCount,omitempty"`
	AvailableCUDAAPIs *AvailableCUDAAPIs `protobuf:"bytes,4,opt,name=availableCUDAAPIs,proto3" json:"availableCUDAAPIs,omitempty"`
	// contains filtered or unexported fields
}

func (*HealthCheckResponse) Descriptor deprecated added in v0.9.211 

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

Deprecated: Use HealthCheckResponse.ProtoReflect.Descriptor instead.

func (*HealthCheckResponse) GetAvailableCUDAAPIs added in v0.9.211 

func (x *HealthCheckResponse) GetAvailableCUDAAPIs() *AvailableCUDAAPIs

func (*HealthCheckResponse) GetDeviceCount added in v0.9.211 

func (x *HealthCheckResponse) GetDeviceCount() int32

func (*HealthCheckResponse) GetSuccess added in v0.9.211 

func (x *HealthCheckResponse) GetSuccess() bool

func (*HealthCheckResponse) GetVersion added in v0.9.211 

func (x *HealthCheckResponse) GetVersion() string

func (*HealthCheckResponse) ProtoMessage added in v0.9.211 

func (*HealthCheckResponse) ProtoMessage()

func (*HealthCheckResponse) ProtoReflect added in v0.9.211 

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

func (*HealthCheckResponse) Reset added in v0.9.211 

func (x *HealthCheckResponse) Reset()

func (*HealthCheckResponse) String added in v0.9.211 

func (x *HealthCheckResponse) String() string

type RestoreRequest 

type RestoreRequest struct {
	Directory string `protobuf:"bytes,1,opt,name=directory,proto3" json:"directory,omitempty"`
	// contains filtered or unexported fields
}

func (*RestoreRequest) Descriptor deprecated 

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

Deprecated: Use RestoreRequest.ProtoReflect.Descriptor instead.

func (*RestoreRequest) GetDirectory added in v0.9.5 

func (x *RestoreRequest) GetDirectory() string

func (*RestoreRequest) ProtoMessage 

func (*RestoreRequest) ProtoMessage()

func (*RestoreRequest) ProtoReflect 

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

func (*RestoreRequest) Reset 

func (x *RestoreRequest) Reset()

func (*RestoreRequest) String 

func (x *RestoreRequest) String() string

type RestoreResponse 

type RestoreResponse struct {
	Success         bool             `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`
	GpuRestoreStats *GPURestoreStats `protobuf:"bytes,2,opt,name=gpuRestoreStats,proto3" json:"gpuRestoreStats,omitempty"`
	// contains filtered or unexported fields
}

func (*RestoreResponse) Descriptor deprecated 

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

Deprecated: Use RestoreResponse.ProtoReflect.Descriptor instead.

func (*RestoreResponse) GetGpuRestoreStats added in v0.9.216 

func (x *RestoreResponse) GetGpuRestoreStats() *GPURestoreStats

func (*RestoreResponse) GetSuccess 

func (x *RestoreResponse) GetSuccess() bool

func (*RestoreResponse) ProtoMessage 

func (*RestoreResponse) ProtoMessage()

func (*RestoreResponse) ProtoReflect 

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

func (*RestoreResponse) Reset 

func (x *RestoreResponse) Reset()

func (*RestoreResponse) String 

func (x *RestoreResponse) String() string

type StartupPollRequest added in v0.9.208 

type StartupPollRequest struct {
	// contains filtered or unexported fields
}

func (*StartupPollRequest) Descriptor deprecated added in v0.9.208 

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

Deprecated: Use StartupPollRequest.ProtoReflect.Descriptor instead.

func (*StartupPollRequest) ProtoMessage added in v0.9.208 

func (*StartupPollRequest) ProtoMessage()

func (*StartupPollRequest) ProtoReflect added in v0.9.208 

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

func (*StartupPollRequest) Reset added in v0.9.208 

func (x *StartupPollRequest) Reset()

func (*StartupPollRequest) String added in v0.9.208 

func (x *StartupPollRequest) String() string

type StartupPollResponse added in v0.9.208 

type StartupPollResponse struct {
	Success bool `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`
	// contains filtered or unexported fields
}

func (*StartupPollResponse) Descriptor deprecated added in v0.9.208 

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

Deprecated: Use StartupPollResponse.ProtoReflect.Descriptor instead.

func (*StartupPollResponse) GetSuccess added in v0.9.208 

func (x *StartupPollResponse) GetSuccess() bool

func (*StartupPollResponse) ProtoMessage added in v0.9.208 

func (*StartupPollResponse) ProtoMessage()

func (*StartupPollResponse) ProtoReflect added in v0.9.208 

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

func (*StartupPollResponse) Reset added in v0.9.208 

func (x *StartupPollResponse) Reset()

func (*StartupPollResponse) String added in v0.9.208 

func (x *StartupPollResponse) String() string

type UnimplementedCedanaGPUServer 

type UnimplementedCedanaGPUServer struct {
}

UnimplementedCedanaGPUServer must be embedded to have forward compatible implementations.

func (UnimplementedCedanaGPUServer) Checkpoint 

func (UnimplementedCedanaGPUServer) Checkpoint(context.Context, *CheckpointRequest) (*CheckpointResponse, error)

func (UnimplementedCedanaGPUServer) HealthCheck added in v0.9.211 

func (UnimplementedCedanaGPUServer) HealthCheck(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error)

func (UnimplementedCedanaGPUServer) Restore 

func (UnimplementedCedanaGPUServer) Restore(context.Context, *RestoreRequest) (*RestoreResponse, error)

func (UnimplementedCedanaGPUServer) StartupPoll added in v0.9.208 

func (UnimplementedCedanaGPUServer) StartupPoll(context.Context, *StartupPollRequest) (*StartupPollResponse, error)

type UnsafeCedanaGPUServer 

type UnsafeCedanaGPUServer interface {
	// contains filtered or unexported methods
}

UnsafeCedanaGPUServer may be embedded to opt out of forward compatibility for this service. Use of this interface is not recommended, as added methods to CedanaGPUServer will result in compilation errors.

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.