Enable zba and zbs for RISCV64 Android

Reviewers: enh, pirama, srhines, asb

Differential Revision: https://reviews.llvm.org/D155339

Author: hiraditya

clang/lib/Driver/ToolChains/Arch/RISCV.cpp

@@ -294,7 +294,7 @@ StringRef riscv::getRISCVArch(const llvm::opt::ArgList &Args,
       return "rv32imafdc";
     else if (MABI.starts_with_insensitive("lp64")) {
       if (Triple.isAndroid())
-        return "rv64imafdc_zbb";
+        return "rv64imafdc_zba_zbb_zbs";
 
       return "rv64imafdc";
     }
@@ -314,7 +314,7 @@ StringRef riscv::getRISCVArch(const llvm::opt::ArgList &Args,
     if (Triple.getOS() == llvm::Triple::UnknownOS)
       return "rv64imac";
     else if (Triple.isAndroid())
-      return "rv64imafdc_zbb";
+      return "rv64imafdc_zba_zbb_zbs";
     else
       return "rv64imafdc";
   }

clang/test/Driver/riscv-features.c

@@ -10,7 +10,9 @@
 // RUN: %clang --target=riscv32-unknown-elf -### %s -mrelax 2>&1 | FileCheck %s -check-prefix=RELAX
 // RUN: %clang --target=riscv32-unknown-elf -### %s -mno-relax 2>&1 | FileCheck %s -check-prefix=NO-RELAX
 
+// ANDROID: "-target-feature" "+zba"
 // ANDROID: "-target-feature" "+zbb"
+// ANDROID: "-target-feature" "+zbs"
 // RELAX: "-target-feature" "+relax"
 // NO-RELAX: "-target-feature" "-relax"
 // DEFAULT: "-target-feature" "+relax"