# COPIED FROM DeepGEMM
def align(x: int, y: int) -> int:
return ceil_div(x, y) * y
def ceil_div(x: int, y: int) -> int:
return (x + y - 1) // y