get_platform

class brainstate.environ.get_platform

Get the current computing platform.

Returns:

Platform name (‘cpu’, ‘gpu’, or ‘tpu’).

Return type:

str

Examples

>>> import brainstate.environ as env
>>>
>>> platform = env.get_platform()
>>> print(f"Running on: {platform}")
>>>
>>> if platform == 'gpu':
...     print("GPU acceleration available")
... else:
...     print(f"Using {platform.upper()}")