def divide(numerator, denominator):
    """Ensure that numerator is divisible by the denominator and return
    the division value."""
    ensure_divisibility(numerator, denominator)
    return numerator // denominator
Ensure that numerator is divisible by the denominator and return
    the division value.
megatron_lm/megatron/mpu/utils.py