Add further support for riscv in prelude and etc (#1009)

Summary: Pull Request resolved: #1009

Reviewed By: scottcao, dtolnay

Differential Revision: D77126594

Pulled By: Nero5023

fbshipit-source-id: c4edf990f8c7a9fc2fe11d1899795ae8a08d55db

Author: xizheyin

buck2.py

@@ -45,6 +45,8 @@ def get_extra_build_params(args: argparse.Namespace) -> List[str]:
         params.extend(["-m", "x86_64"])
     elif arch_platform == "arm64":
         params.extend(["-m", "arm64"])
+    elif arch_platform == "riscv64":
+        params.extend(["-m", "riscv64"])
 
     return params
 

examples/bootstrap/bootstrap/platform/platforms.bzl

@@ -49,6 +49,8 @@ def _host_cpu_configuration() -> str:
         return "prelude//cpu:arm32"
     elif arch.is_i386:
         return "prelude//cpu:x86_32"
+    elif arch.is_riscv64:
+        return "prelude//cpu:riscv64"
     else:
         return "prelude//cpu:x86_64"
 

examples/bootstrap/toolchains/rust_toolchain.bzl

@@ -14,6 +14,7 @@ load("@prelude//rust:rust_toolchain.bzl", "PanicRuntime", "RustToolchainInfo")
 _DEFAULT_TRIPLE = select({
     "config//os:linux": select({
         "config//cpu:arm64": "aarch64-unknown-linux-gnu",
+        "config//cpu:riscv64": "riscv64gc-unknown-linux-gnu",
         "config//cpu:x86_64": "x86_64-unknown-linux-gnu",
     }),
     "config//os:macos": select({

examples/persistent_worker/platforms/BUCK

@@ -3,7 +3,9 @@ load(":local.bzl", "local")
 
 oncall("build_infra")
 
-host_cpu = "prelude//cpu:" + ("arm64" if host_info().arch.is_aarch64 else "x86_64")
+host_cpu = "prelude//cpu:" + (
+    "arm64" if host_info().arch.is_aarch64 else "riscv64" if host_info().arch.is_riscv64 else "x86_64"
+)
 
 host_os = "prelude//os:" + ("macos" if host_info().os.is_macos else "linux")
 

prelude/cpu/BUCK

@@ -43,3 +43,11 @@ config_setting(
     ],
     visibility = ["PUBLIC"],
 )
+
+config_setting(
+    name = "riscv64",
+    constraint_values = [
+        "prelude//cpu/constraints:riscv64",
+    ],
+    visibility = ["PUBLIC"],
+)

prelude/cpu/constraints/BUCK

@@ -48,3 +48,10 @@ constraint_value(
     constraint_setting = ":cpu",
     visibility = ["PUBLIC"],
 )
+
+# RISC-V 64-bit support
+constraint_value(
+    name = "riscv64",
+    constraint_setting = ":cpu",
+    visibility = ["PUBLIC"],
+)

prelude/decls/core_rules.bzl

@@ -21,7 +21,7 @@ PlatformExePlatform = ["linux", "macos", "windows"]
 
 RemoteFileType = ["data", "executable", "exploded_zip"]
 
-TargetCpuType = ["arm", "armv7", "arm64", "x86", "x86_64", "mips"]
+TargetCpuType = ["arm", "armv7", "arm64", "x86", "x86_64", "mips", "riscv64"]
 
 alias = prelude_rule(
     name = "alias",

prelude/os_lookup/targets/BUCK

@@ -10,6 +10,7 @@ os_lookup(
     cpu = select({
         "DEFAULT": None,
         "config//cpu:arm64": "arm64",
+        "config//cpu:riscv64": "riscv64",
         "config//cpu:x86_64": "x86_64",
     }),
     os = select({

prelude/platforms/defs.bzl

@@ -47,6 +47,8 @@ def _host_cpu_configuration() -> str:
         return "prelude//cpu:arm32"
     elif arch.is_i386:
         return "prelude//cpu:x86_32"
+    elif arch.is_riscv64:
+        return "prelude//cpu:riscv64"
     else:
         return "prelude//cpu:x86_64"
 

prelude/rust/cargo_package.bzl

@@ -18,6 +18,7 @@ DEFAULT_PLATFORM_TEMPLATES = select({
     "config//os:linux": select({
         "DEFAULT": None,
         "config//cpu:arm64": "linux-arm64",
+        "config//cpu:riscv64": "linux-riscv64",
         "config//cpu:x86_64": "linux-x86_64",
     }),
     "config//os:macos": select({