gpu

package

v0.0.0-...-6a277b3 Latest Latest Go to latest Published: Feb 1, 2025 License: MIT Imports: 0 Imported by: 0

Documentation ¶

This section is empty.

This section is empty.

This section is empty.

type Device struct{}

Device describes a GPU computing node.

func NewDevice() *Device

NewDevice returns a new GPU-backed device instance.

func (*Device) Square(in []float32) []float32

Square implements the compute.DeviceDevice interface.

func (*Device) Sum(in []float32) float32

Sum implements the compute.DeviceDevice interface.