def METHOD_NAME(
    self, *, dim, displacement, courant, cell_origin, position_in_cell, n_substeps
):
    n_dim = len(courant.shape)
    n_sd = position_in_cell.shape[1]
    self.__calculate_displacement_body[n_dim].launch_n(
        n=n_sd,
        args=(
            trtc.DVInt64(dim),
            trtc.DVInt64(n_sd),
            displacement.data,
            courant.data,
            trtc.DVInt64(courant.shape[0]),
            trtc.DVInt64(courant.shape[1] if n_dim > 2 else -1),
            cell_origin.data,
            position_in_cell.data,
            trtc.DVInt64(n_substeps),
        ),
    )