mm: hugetlb: Add huge page size param to set_huge_pte_at()

In order to fix a bug, arm64 needs to be told the size of the huge page
for which the pte is being set in set_huge_pte_at(). Provide for this by
adding an `unsigned long sz` parameter to the function. This follows the
same pattern as huge_pte_clear().

This commit makes the required interface modifications to the core mm as
well as all arches that implement this function (arm64, parisc, powerpc,
riscv, s390, sparc). The actual arm64 bug will be fixed in a separate
commit.

No behavioral changes intended.

Signed-off-by: Ryan Roberts <ryan.roberts@arm.com>
Cc: <stable@vger.kernel.org> # 6.5+