Skip to main content

Mojo struct

CommTuningConfig

struct CommTuningConfig

Parameters: ngpus: Number of GPUs for running allreduce. num_bytes: Total number of input bytes supported by the config. sm_version: SM version (as string). num_blocks: Number of thread blocks for running allreduce.

Fields​

  • ​ngpus (Int):
  • ​num_bytes (Int):
  • ​sm_version (StaticString):
  • ​num_blocks (Int):

Implemented traits​

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDestructible, Movable, RegisterPassable, TrivialRegisterPassable, TuningConfig, Writable

Methods​

write_to​

write_to(self, mut writer: T)

Writes the tuning config as a string.

Args:

  • ​writer (T): The writer to write to.