def METHOD_NAME(self) -> Optional[QuantizationContext]:
    # fp8 rowwise is activated when row_dim > 0
    if self._comm_precision == SparseType.FP8:
        return QuantizationContext(self._row_dim)
    # int8 rowwise is default
    return QuantizationContext()