Python class

DeviceSpec

DeviceSpec

class max.driver.DeviceSpec(id, device_type='cpu')

source

Bases: object

A device specification containing an ID and device type.

Parameters:

• id (int) – The device identifier.
• device_type (Literal['cpu', 'gpu']) – The device type, either cpu or gpu. Defaults to cpu.

accelerator()

static accelerator(id=0)

source

Creates an accelerator (GPU) device specification.

Parameters:

id (int)

cpu()

static cpu(id=-1)

source

Creates a CPU device specification.

Parameters:

id (int)

device_type

device_type: Literal['cpu', 'gpu'] = 'cpu'

source

The device type, either cpu or gpu.

id

id: int

source

The device identifier.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit