Skip to main content

Mojo struct

TuningConfigAllreduce

struct TuningConfigAllreduce

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.

Was this page helpful?