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
__str__
__str__(self) -> String
Returns:
String
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?
Thank you! We'll create more content like this.
Thank you for helping us improve!