types

type CPUArchitectureEnum string

type GPUNodeInstanceInfo struct {
	InstanceType string
	CPUs         int32
	MemoryGiB    int32

	// In TFlops and GB for every GPU
	FP16TFlopsPerGPU    float64
	VRAMGigabytesPerGPU int32

	GPUModel string
	GPUCount int32

	CPUArchitecture CPUArchitectureEnum
	GPUVendor       GPUVendorEnum
}

type GPUNodeProvider interface {
	TestConnection() error

	CreateNode(ctx context.Context, param *tfv1.GPUNodeClaim) (*GPUNodeStatus, error)
	TerminateNode(ctx context.Context, param *NodeIdentityParam) error
	GetNodeStatus(ctx context.Context, param *NodeIdentityParam) (*GPUNodeStatus, error)

	GetInstancePricing(instanceType string, capacityType tfv1.CapacityTypeEnum, region string) (float64, error)

	GetGPUNodeInstanceTypeInfo(region string) []GPUNodeInstanceInfo
}

type GPUNodeStatus struct {
	InstanceID string
	CreatedAt  time.Time

	PrivateIP string
	PublicIP  string
}

type GPUVendorEnum string

type NodeIdentityParam struct {
	InstanceID string
	Region     string
}