ffu

type AESCapability struct {
	// Supported key sizes in bits
	KeySizes []int

	// Supported modes
	Modes []AESMode

	// Maximum throughput in MB/s
	ThroughputMBps map[AESMode]float64

	// Hardware acceleration available
	HardwareAccelerated bool

	// Supports parallel operations
	SupportsParallel bool

	// Maximum parallel operations
	MaxParallel int
}

type AESMode int

type AESOperation int

type AESWorkload struct {
	Operation AESOperation
	Mode      AESMode
	Key       []byte // 128, 192, or 256 bit
	IV        []byte // Initialization vector (for modes that need it)
	Input     []byte // Plaintext or ciphertext
	Output    []byte // Result buffer (must be pre-allocated)

	// Optional: Additional authenticated data for GCM mode
	AAD []byte
}

type AMXCapability struct {
	// Supported data types
	SupportsInt8 bool
	SupportsBF16 bool
	SupportsFP16 bool

	// Tile configuration
	NumTiles     int
	MaxTileRows  int
	MaxTileCols  int
	MaxTileBytes int

	// Performance characteristics
	PeakInt8TOPS   float64 // Peak INT8 operations per second
	PeakBF16TFLOPS float64 // Peak BF16 operations per second

	// Constraints
	RequiresAlignment int // Byte alignment requirement
}

type AMXDataType int

type AMXOperation int

type AMXWorkload struct {
	Operation AMXOperation
	DataType  AMXDataType

	// Matrix dimensions
	M int // Rows of A and C
	N int // Columns of B and C
	K int // Columns of A, Rows of B

	// Matrix data
	A []byte // M×K matrix (row-major)
	B []byte // K×N matrix (row-major)
	C []byte // M×N matrix (row-major)

	// Scaling factors for quantized operations
	ScaleA float32 // Scale factor for A
	ScaleB float32 // Scale factor for B
	ScaleC float32 // Scale factor for C

}

type Cost struct {
	// Estimated execution time
	Duration time.Duration

	// Estimated energy usage in joules
	Energy float64

	// Estimated memory bandwidth usage in bytes
	MemoryBandwidth int64

	// Confidence level (0-1) in the estimate
	Confidence float64
}

type FFU interface {
	// Name returns the FFU name
	Name() string

	// Type returns the FFU type
	Type() FFUType

	// CanHandle checks if this FFU can handle the workload
	CanHandle(workload Workload) bool

	// EstimateCost estimates the cost of executing the workload
	EstimateCost(workload Workload) Cost

	// Execute runs the workload on this FFU
	Execute(workload Workload) error

	// IsAvailable checks if the FFU is currently available
	IsAvailable() bool

	// Metrics returns performance metrics for this FFU
	Metrics() Metrics
}

type FFUType int

type Metrics struct {
	// Total number of workloads executed
	WorkloadCount int64

	// Total bytes processed
	BytesProcessed int64

	// Total execution time
	TotalDuration time.Duration

	// Total energy consumed (if available)
	TotalEnergy float64

	// Number of errors
	ErrorCount int64

	// Last error (if any)
	LastError error

	// Timestamp of last use
	LastUsed time.Time
}

type PCLMULOperation int

type PCLMULWorkload struct {
	Operation  PCLMULOperation
	Data       []byte    // Input data
	Output     []byte    // Output buffer
	Polynomial uint64    // For CRC32
	Result     uint64    // For CRC32 result
	RSConfig   *RSConfig // For Reed-Solomon
}

type RSConfig struct {
	DataShards   int
	ParityShards int
}

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

type SHAAlgorithm int

type SHAWorkload struct {
	Algorithm SHAAlgorithm
	Data      []byte
	Output    []byte // Pre-allocated output buffer
}

type VNNIOperation int

type VNNIWorkload struct {
	Operation VNNIOperation
	M, N, K   int     // Matrix dimensions
	A         []int8  // M×K matrix
	B         []int8  // K×N matrix
	C         []int32 // M×N matrix (output)
	Alpha     int32   // Scaling factor
}

type Workload interface {
	// Type returns the workload type (e.g., "crypto_aes", "matmul_int8")
	Type() string

	// Size returns the size of the workload in bytes
	Size() int64

	// Validate checks if the workload is valid
	Validate() error
}