auto_select

hoomd.device.auto_select(communicator=None, message_filename=None, notice_level=2)

Automatically select the hardware device.

Parameters:

  • communicator (hoomd.communicator.Communicator) – MPI communicator object. When None, create a default communicator that uses all MPI ranks.

  • message_filename (str) – Filename to write messages to. When None, use sys.stdout and sys.stderr. Messages from multiple MPI ranks are collected into this file.

  • notice_level (int) – Minimum level of messages to print.

Returns:

Instance of GPU if available, otherwise CPU.

Example:

device = hoomd.device.auto_select()