Enable build on aarch64 linux (#3896) (#4017)

Summary:
There's a mismatch in torch and torchvision dependecies for the linux-aarch64 packages and missing support in resolve_buck script.

Signed-off-by: Per Åstrand <[email protected]>

Change-Id: I491499ca5e524fd2788919b6446a370fe44fdb86

Pull Request resolved: #3896

Reviewed By: digantdesai

Differential Revision: D58741803

Pulled By: mergennachin

fbshipit-source-id: 7fe598da58ea6fc29726f38cfb394a9eda832c44
(cherry picked from commit 337174c)

Co-authored-by: Per Åstrand <[email protected]>

Author: pytorchbot

build/resolve_buck.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # Copyright (c) Meta Platforms, Inc. and affiliates.
+# Copyright 2024 Arm Limited and/or its affiliates.
 # All rights reserved.
 #
 # This source code is licensed under the BSD-style license found in the
@@ -72,6 +73,10 @@ class BuckInfo:
             "029b0bcc6f8e399185c1d0f574eba204934722b5",
         ],
     ),
+    ("linux", "aarch64"): BuckInfo(
+        archive_name="buck2-aarch64-unknown-linux-gnu.zst",
+        target_versions=["49670bee56a7d8a7696409ca6fbf7551d2469787"],
+    ),
     ("darwin", "aarch64"): BuckInfo(
         archive_name="buck2-aarch64-apple-darwin.zst",
         target_versions=["99773fe6f7963a72ae5f7b737c02836e"],
@@ -122,7 +127,7 @@ def resolve_buck2(args: argparse.Namespace) -> Union[str, int]:
     arch = "unknown"
     if machine == "x86" or machine == "x86_64" or machine == "amd64":
         arch = "x86_64"
-    elif machine == "arm64":
+    elif machine == "arm64" or machine == "aarch64":
         arch = "aarch64"
 
     os_family = "unknown"