gpu

package
v0.0.0-...-9b45271 Latest Latest
Warning

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

 Go to latest Published: Sep 3, 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 

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 

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

Deprecated: Use AvailableCUDAAPIs.ProtoReflect.Descriptor instead.

func (*AvailableCUDAAPIs) GetCuBLASVersion 

func (x *AvailableCUDAAPIs) GetCuBLASVersion() int32

func (*AvailableCUDAAPIs) GetCuDNNVersion 

func (x *AvailableCUDAAPIs) GetCuDNNVersion() int32

func (*AvailableCUDAAPIs) GetDriverVersion 

func (x *AvailableCUDAAPIs) GetDriverVersion() int32

func (*AvailableCUDAAPIs) GetNcclVersion 

func (x *AvailableCUDAAPIs) GetNcclVersion() int32

func (*AvailableCUDAAPIs) GetRuntimeVersion 

func (x *AvailableCUDAAPIs) GetRuntimeVersion() int32

func (*AvailableCUDAAPIs) ProtoMessage 

func (*AvailableCUDAAPIs) ProtoMessage()

func (*AvailableCUDAAPIs) ProtoReflect 

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

func (*AvailableCUDAAPIs) Reset 

func (x *AvailableCUDAAPIs) Reset()

func (*AvailableCUDAAPIs) String 

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 

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 

func (x *CheckpointResponse) GetCkptPath() string

func (*CheckpointResponse) GetMemPath 

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 

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 

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

Deprecated: Use GPURestoreStats.ProtoReflect.Descriptor instead.

func (*GPURestoreStats) GetCopyMemTime 

func (x *GPURestoreStats) GetCopyMemTime() int64

func (*GPURestoreStats) GetReplayCallsTime 

func (x *GPURestoreStats) GetReplayCallsTime() int64

func (*GPURestoreStats) ProtoMessage 

func (*GPURestoreStats) ProtoMessage()

func (*GPURestoreStats) ProtoReflect 

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

func (*GPURestoreStats) Reset 

func (x *GPURestoreStats) Reset()

func (*GPURestoreStats) String 

func (x *GPURestoreStats) String() string

type HealthCheckRequest 

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

func (*HealthCheckRequest) Descriptor deprecated 

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

Deprecated: Use HealthCheckRequest.ProtoReflect.Descriptor instead.

func (*HealthCheckRequest) ProtoMessage 

func (*HealthCheckRequest) ProtoMessage()

func (*HealthCheckRequest) ProtoReflect 

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

func (*HealthCheckRequest) Reset 

func (x *HealthCheckRequest) Reset()

func (*HealthCheckRequest) String 

func (x *HealthCheckRequest) String() string

type HealthCheckResponse 

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 

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

Deprecated: Use HealthCheckResponse.ProtoReflect.Descriptor instead.

func (*HealthCheckResponse) GetAvailableCUDAAPIs 

func (x *HealthCheckResponse) GetAvailableCUDAAPIs() *AvailableCUDAAPIs

func (*HealthCheckResponse) GetDeviceCount 

func (x *HealthCheckResponse) GetDeviceCount() int32

func (*HealthCheckResponse) GetSuccess 

func (x *HealthCheckResponse) GetSuccess() bool

func (*HealthCheckResponse) GetVersion 

func (x *HealthCheckResponse) GetVersion() string

func (*HealthCheckResponse) ProtoMessage 

func (*HealthCheckResponse) ProtoMessage()

func (*HealthCheckResponse) ProtoReflect 

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

func (*HealthCheckResponse) Reset 

func (x *HealthCheckResponse) Reset()

func (*HealthCheckResponse) String 

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 

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 

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 

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

func (*StartupPollRequest) Descriptor deprecated 

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

Deprecated: Use StartupPollRequest.ProtoReflect.Descriptor instead.

func (*StartupPollRequest) ProtoMessage 

func (*StartupPollRequest) ProtoMessage()

func (*StartupPollRequest) ProtoReflect 

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

func (*StartupPollRequest) Reset 

func (x *StartupPollRequest) Reset()

func (*StartupPollRequest) String 

func (x *StartupPollRequest) String() string

type StartupPollResponse 

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

func (*StartupPollResponse) Descriptor deprecated 

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

Deprecated: Use StartupPollResponse.ProtoReflect.Descriptor instead.

func (*StartupPollResponse) GetSuccess 

func (x *StartupPollResponse) GetSuccess() bool

func (*StartupPollResponse) ProtoMessage 

func (*StartupPollResponse) ProtoMessage()

func (*StartupPollResponse) ProtoReflect 

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

func (*StartupPollResponse) Reset 

func (x *StartupPollResponse) Reset()

func (*StartupPollResponse) String 

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 

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

func (UnimplementedCedanaGPUServer) Restore 

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

func (UnimplementedCedanaGPUServer) StartupPoll 

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.