Skip to main content

//clika-runtime/io.clika.runtime/Device

Device

[jvm]
data class Device(val api: Int, val index: Int = 0)

A compute placement: the device= value the factories and movers take.

val cpu = Device.CPU
val gpu0 = Device.parse("cuda:0")
val t = Tensors.zeros(longArrayOf(2, 3), device = gpu0)

Constructors

Device[jvm]
constructor(api: Int, index: Int = 0)

Types

NameSummary
Companion[jvm]
object Companion

Properties

NameSummary
api[jvm]
val api: Int
index[jvm]
val index: Int = 0

Functions

NameSummary
toString[jvm]
open override fun toString(): String