Override to resolve 'auto' values to dtype field. When a plugin field has value 'auto', return the value of dtype instead.
(self, name: str)
| 295 | "ReduceScatter + AllGather targeting large MoE models.") |
| 296 | |
| 297 | def __getattribute__(self, name: str) -> Any: |
| 298 | """Override to resolve 'auto' values to dtype field. |
| 299 | |
| 300 | When a plugin field has value 'auto', return the value of dtype instead. |
| 301 | """ |
| 302 | # Use object.__getattribute__ to avoid infinite recursion |
| 303 | value = object.__getattribute__(self, name) |
| 304 | |
| 305 | if name != "dtype" and value == "auto": |
| 306 | return self.dtype |
| 307 | |
| 308 | return value |
| 309 | |
| 310 | @field_validator("dtype") |
| 311 | @classmethod |
no outgoing calls
no test coverage detected