(
self,
name: str,
operand_dtype,
scale_dtype,
acc_dtype=Float32,
scale_factor=None,
)
| 695 | """Base class for scaled MMA configuration.""" |
| 696 | |
| 697 | def __init__( |
| 698 | self, |
| 699 | name: str, |
| 700 | operand_dtype, |
| 701 | scale_dtype, |
| 702 | acc_dtype=Float32, |
| 703 | scale_factor=None, |
| 704 | ): |
| 705 | self.name = name |
| 706 | self.operand_dtype = operand_dtype |
| 707 | self.scale_dtype = scale_dtype |
| 708 | self.acc_dtype = acc_dtype |
| 709 | # Default: 16 for e4m3 scale (4X / mxf4nvf4), 32 otherwise (2X / mxf4, mxf8f6f4) |
| 710 | if scale_factor is None: |
| 711 | scale_factor = 16 if scale_dtype == Float8E4M3FN else 32 |
| 712 | self.scale_factor = scale_factor |
| 713 | |
| 714 | def __str__(self): |
| 715 | return self.name |
nothing calls this directly
no outgoing calls
no test coverage detected