ttnn.SetDefaultDevice

ttnn.SetDefaultDevice(arg0: ttnn._ttnn.device.IDevice) → None

Sets the default device to use for operations when inputs are not on the device.

Parameters:

device (ttnn.Device) – The TT device to use.

Note

This functionality is planned for deprecation in the future.

Example

>>> device_id = 0
>>> device = ttnn.open_device(device_id = device_id)
>>> ttnn.SetDefaultDevice(device)