net.thevpc.nuts.runtime.standalone.platform

Class DefaultNGpuDevice

java.lang.Object

net.thevpc.nuts.runtime.standalone.platform.DefaultNGpuDevice

All Implemented Interfaces:

NGpuDevice

public class DefaultNGpuDevice
extends Object
implements NGpuDevice

Default implementation of NGpuDevice.

Since:

1.0.0

Author:

thevpc

Field Summary

Fields inherited from interface net.thevpc.nuts.platform.NGpuDevice

CC_AMPERE, CC_DP4A, CC_PASCAL, CC_TURING, COMPUTE_CAPABILITY, DEVICE_TYPE, DRIVER_VERSION, KERNEL_DRIVER, MEMORY_BANDWIDTH_GBPS, PCI_BUS_ID, PCI_DEVICE_ID, PCIE_GEN_CURRENT, PCIE_GEN_MAX, PCIE_WIDTH_CURRENT, PCIE_WIDTH_MAX, PRIMARY_GPU_PROPERTY, UUID, VENDOR

Constructor Summary

Constructors

Constructor and Description
DefaultNGpuDevice(String name, NRam vram, Map<String,String> capabilities)

Method Summary

Modifier and Type	Method and Description
Map<String,String>	capabilities()
NOptional<String>	capability(String key)
int	computeCapability()
NGpuDeviceType	deviceType()
boolean	equals(Object o)
int	hashCode()
boolean	hasTensorCores()
boolean	isComputeCapable()
boolean	isSupportedBf16()
boolean	isSupportedFp16()
boolean	isSupportedInt4()
boolean	isSupportedInt8()
String	name()
static List<NGpuDevice>	orderWithPrimaryFirst(List<NGpuDevice> gpus)
String	toString()
NGpuVendor	vendor()
NRam	vram()

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface net.thevpc.nuts.platform.NGpuDevice

primary

Constructor Detail

DefaultNGpuDevice

public DefaultNGpuDevice(String name,
                         NRam vram,
                         Map<String,String> capabilities)

Method Detail

name

public String name()

Specified by:

name in interface NGpuDevice

vram

public NRam vram()

Specified by:

vram in interface NGpuDevice

capability

public NOptional<String> capability(String key)

Specified by:

capability in interface NGpuDevice

capabilities

public Map<String,String> capabilities()

Specified by:

capabilities in interface NGpuDevice

vendor

public NGpuVendor vendor()

Specified by:

vendor in interface NGpuDevice

deviceType

public NGpuDeviceType deviceType()

Specified by:

deviceType in interface NGpuDevice

computeCapability

public int computeCapability()

Specified by:

computeCapability in interface NGpuDevice

isComputeCapable

public boolean isComputeCapable()

Specified by:

isComputeCapable in interface NGpuDevice

hasTensorCores

public boolean hasTensorCores()

Specified by:

hasTensorCores in interface NGpuDevice

isSupportedFp16

public boolean isSupportedFp16()

Specified by:

isSupportedFp16 in interface NGpuDevice

isSupportedBf16

public boolean isSupportedBf16()

Specified by:

isSupportedBf16 in interface NGpuDevice

isSupportedInt8

public boolean isSupportedInt8()

Specified by:

isSupportedInt8 in interface NGpuDevice

isSupportedInt4

public boolean isSupportedInt4()

Specified by:

isSupportedInt4 in interface NGpuDevice

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object

orderWithPrimaryFirst

public static List<NGpuDevice> orderWithPrimaryFirst(List<NGpuDevice> gpus)

toString

public String toString()

Overrides:

toString in class Object