tools: py: add riscv64 to pip and manylinux compilance test

Author: infiWang

tensorflow/tools/pip_package/BUILD

@@ -307,6 +307,7 @@ tf_wheel(
         "@platforms//cpu:arm64": "arm64",
         "@platforms//cpu:x86_64": "x86_64",
         "@platforms//cpu:ppc": "ppc64le",
+        "@platforms//cpu:riscv64": "riscv64",
     }),
     source_files = [
         "MANIFEST.in",
@@ -422,6 +423,7 @@ verify_manylinux_compliance_test(
     name = "manylinux_compliance_test",
     aarch64_compliance_tag = "manylinux_2_17_aarch64",
     ppc64le_compliance_tag = "manylinux_2_17_ppc64le",
+    riscv64_compliance_tag = "manylinux_2_17_riscv64",
     test_tags = [
         "manual",
     ],

third_party/xla/third_party/py/manylinux_compliance_test.py

@@ -46,6 +46,11 @@ def parse_args():
       required=True,
       help="ManyLinux compliance tag for ppc64le",
   )
+  parser.add_argument(
+      "--riscv64-compliance-tag",
+      required=True,
+      help="ManyLinux compliance tag for riscv64",
+  )
   return parser.parse_args()
 
 
@@ -106,7 +111,7 @@ def verify_manylinux_compliance(
 
 def test_manylinux_compliance(args):
   machine_type = platform.uname().machine
-  supported_machine_types = ["x86_64", "aarch64", "ppc64le"]
+  supported_machine_types = ["x86_64", "aarch64", "ppc64le", "riscv64"]
   if machine_type not in supported_machine_types:
     raise RuntimeError(
         "Unsupported machine type {machine_type}. The supported are:"
@@ -118,8 +123,10 @@ def test_manylinux_compliance(args):
     compliance_tag = args.x86_64_compliance_tag
   elif machine_type == "aarch64":
     compliance_tag = args.aarch64_compliance_tag
-  else:
+  elif machine_type == "ppc64le":
     compliance_tag = args.ppc64le_compliance_tag
+  else:  # machine_type == "riscv64"
+    compliance_tag = args.riscv64_compliance_tag
   auditwheel_output = get_auditwheel_output(args.wheel_path)
   verify_manylinux_compliance(
       auditwheel_output,

third_party/xla/third_party/py/py_manylinux_compliance_test.bzl

@@ -8,6 +8,7 @@ def verify_manylinux_compliance_test(
         aarch64_compliance_tag,
         x86_64_compliance_tag,
         ppc64le_compliance_tag,
+        riscv64_compliance_tag,
         test_tags = []):
     py_test(
         name = name,
@@ -21,6 +22,7 @@ def verify_manylinux_compliance_test(
             "--aarch64-compliance-tag={}".format(aarch64_compliance_tag),
             "--x86_64-compliance-tag={}".format(x86_64_compliance_tag),
             "--ppc64le-compliance-tag={}".format(ppc64le_compliance_tag),
+            "--riscv64-compliance-tag={}".format(riscv64_compliance_tag),
         ],
         main = "manylinux_compliance_test.py",
         tags = ["manual"] + test_tags,