Device

(source)

`Device` management. A `Device` encapsulates the graph compiler and hardware support.

Beyond the *type* `Device` itself, you will not likely need this module unless you are
developing a custom plugin.

Reexports

import public Compiler.MLIR.IR.DialectRegistry
import public Compiler.MLIR.IR.MLIRContext
import public Compiler.MLIR.Pass.PassManager

Definitions

record Device : Type

  A PJRT "device". These are required to run spidr graphs, and are provided by your plugin.
  
  Plugin developers should create one via `createDevice`. Everyone else should use their plugin.

Totality: total
Visibility: public export
Constructor:

MkDevice : MLIRContext -> PassManager -> PjrtApi -> PjrtClient -> Device

Projections:

.api : Device -> PjrtApi
.client : Device -> PjrtClient
.mlirCtx : Device -> MLIRContext
.passManager : Device -> PassManager

.mlirCtx : Device -> MLIRContext

Totality: total
Visibility: public export

mlirCtx : Device -> MLIRContext

Totality: total
Visibility: public export

.passManager : Device -> PassManager

Totality: total
Visibility: public export

passManager : Device -> PassManager

Totality: total
Visibility: public export

.api : Device -> PjrtApi

Totality: total
Visibility: public export

api : Device -> PjrtApi

Totality: total
Visibility: public export

.client : Device -> PjrtClient

Totality: total
Visibility: public export

client : Device -> PjrtClient

Totality: total
Visibility: public export

createDevice : PjrtApi -> Pjrt Device

  Create and initialize a `Device`. For use by plugin developers.

Totality: total
Visibility: export