get_host_device_count

class brainstate.environ.get_host_device_count

Get the number of host devices.

Returns:

Number of host devices configured.

Return type:

int

Examples

>>> import brainstate.environ as env
>>>
>>> # Get device count
>>> n_devices = env.get_host_device_count()
>>> print(f"Host devices: {n_devices}")
>>>
>>> # Configure for parallel computation
>>> if n_devices > 1:
...     print(f"Can use {n_devices} devices for parallel computation")