[clang] [CIR][RISCV] Support zbkx builitin codegen (PR #195211)

[via cfe-commits]

llvmorg-github-actions[bot] wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-backend-risc-v

Author: Jianjian Guan (jacquesguan)

<details>
<summary>Changes</summary>

Include 4 builtins: __builtin_riscv_xperm4_32, __builtin_riscv_xperm4_64, __builtin_riscv_xperm8_32, __builtin_riscv_xperm8_64.

---
Full diff: https://github.com/llvm/llvm-project/pull/195211.diff


2 Files Affected:

- (modified) clang/lib/CIR/CodeGen/CIRGenBuiltinRISCV.cpp (+8-2) 
- (added) clang/test/CIR/CodeGenBuiltins/RISCV/riscv-zbkx.c (+64) 


``````````diff
diff --git a/clang/lib/CIR/CodeGen/CIRGenBuiltinRISCV.cpp b/clang/lib/CIR/CodeGen/CIRGenBuiltinRISCV.cpp
index 52b666fc30d00..be861fe1eb90e 100644
--- a/clang/lib/CIR/CodeGen/CIRGenBuiltinRISCV.cpp
+++ b/clang/lib/CIR/CodeGen/CIRGenBuiltinRISCV.cpp
@@ -71,9 +71,15 @@ CIRGenFunction::emitRISCVBuiltinExpr(unsigned builtinID, const CallExpr *e) {
 
   // Zbkx
   case RISCV::BI__builtin_riscv_xperm4_32:
-  case RISCV::BI__builtin_riscv_xperm4_64:
+  case RISCV::BI__builtin_riscv_xperm4_64: {
+    intrinsicName = "riscv.xperm4";
+    break;
+  }
   case RISCV::BI__builtin_riscv_xperm8_32:
-  case RISCV::BI__builtin_riscv_xperm8_64:
+  case RISCV::BI__builtin_riscv_xperm8_64: {
+    intrinsicName = "riscv.xperm8";
+    break;
+  }
   // Zbkb
   case RISCV::BI__builtin_riscv_brev8_32:
   case RISCV::BI__builtin_riscv_brev8_64:
diff --git a/clang/test/CIR/CodeGenBuiltins/RISCV/riscv-zbkx.c b/clang/test/CIR/CodeGenBuiltins/RISCV/riscv-zbkx.c
new file mode 100644
index 0000000000000..677f94c8f579e
--- /dev/null
+++ b/clang/test/CIR/CodeGenBuiltins/RISCV/riscv-zbkx.c
@@ -0,0 +1,64 @@
+// RUN: %clang_cc1 -triple riscv32 -target-feature +zbkx -fclangir -emit-cir %s -o - | FileCheck %s --check-prefix=CIR32
+// RUN: %clang_cc1 -triple riscv64 -target-feature +zbkx -fclangir -emit-cir %s -o - | FileCheck %s --check-prefix=CIR64
+// RUN: %clang_cc1 -triple riscv32 -target-feature +zbkx -fclangir -emit-llvm %s -o - | FileCheck %s --check-prefix=LLVM32
+// RUN: %clang_cc1 -triple riscv64 -target-feature +zbkx -fclangir -emit-llvm %s -o - | FileCheck %s --check-prefix=LLVM64
+// RUN: %clang_cc1 -triple riscv32 -target-feature +zbkx -emit-llvm %s -o - | FileCheck %s --check-prefix=OGCG32
+// RUN: %clang_cc1 -triple riscv64 -target-feature +zbkx -emit-llvm %s -o - | FileCheck %s --check-prefix=OGCG64
+
+#if __riscv_xlen == 32
+// CIR32-LABEL: cir.func{{.*}} @test_builtin_xperm4_32(
+// CIR32: {{%.*}} = cir.call_llvm_intrinsic "riscv.xperm4" {{%.*}}, {{%.*}} : (!u32i, !u32i) -> !u32i
+// CIR32: cir.return
+// LLVM32-LABEL: @test_builtin_xperm4_32(
+// LLVM32: call i32 @llvm.riscv.xperm4.i32(i32 {{%.*}}, i32 {{%.*}})
+// LLVM32: ret i32
+// OGCG32-LABEL: @test_builtin_xperm4_32(
+// OGCG32: call i32 @llvm.riscv.xperm4.i32(i32 {{%.*}}, i32 {{%.*}})
+// OGCG32: ret i32
+unsigned int test_builtin_xperm4_32(unsigned int a, unsigned int b) {
+  return __builtin_riscv_xperm4_32(a, b);
+}
+
+// CIR32-LABEL: cir.func{{.*}} @test_builtin_xperm8_32(
+// CIR32: {{%.*}} = cir.call_llvm_intrinsic "riscv.xperm8" {{%.*}}, {{%.*}} : (!u32i, !u32i) -> !u32i
+// CIR32: cir.return
+// LLVM32-LABEL: @test_builtin_xperm8_32(
+// LLVM32: call i32 @llvm.riscv.xperm8.i32(i32 {{%.*}}, i32 {{%.*}})
+// LLVM32: ret i32
+// OGCG32-LABEL: @test_builtin_xperm8_32(
+// OGCG32: call i32 @llvm.riscv.xperm8.i32(i32 {{%.*}}, i32 {{%.*}})
+// OGCG32: ret i32
+unsigned int test_builtin_xperm8_32(unsigned int a, unsigned int b) {
+  return __builtin_riscv_xperm8_32(a, b);
+}
+#endif
+
+#if __riscv_xlen == 64
+// CIR64-LABEL: cir.func{{.*}} @test_builtin_xperm4_64(
+// CIR64: {{%.*}} = cir.call_llvm_intrinsic "riscv.xperm4" {{%.*}}, {{%.*}} : (!u64i, !u64i) -> !u64i
+// CIR64: cir.return
+// LLVM64-LABEL: @test_builtin_xperm4_64(
+// LLVM64: call i64 @llvm.riscv.xperm4.i64(i64 {{%.*}}, i64 {{%.*}})
+// LLVM64: ret i64
+// OGCG64-LABEL: @test_builtin_xperm4_64(
+// OGCG64: call i64 @llvm.riscv.xperm4.i64(i64 {{%.*}}, i64 {{%.*}})
+// OGCG64: ret i64
+unsigned long long test_builtin_xperm4_64(unsigned long long a,
+                                          unsigned long long b) {
+  return __builtin_riscv_xperm4_64(a, b);
+}
+
+// CIR64-LABEL: cir.func{{.*}} @test_builtin_xperm8_64(
+// CIR64: {{%.*}} = cir.call_llvm_intrinsic "riscv.xperm8" {{%.*}}, {{%.*}} : (!u64i, !u64i) -> !u64i
+// CIR64: cir.return
+// LLVM64-LABEL: @test_builtin_xperm8_64(
+// LLVM64: call i64 @llvm.riscv.xperm8.i64(i64 {{%.*}}, i64 {{%.*}})
+// LLVM64: ret i64
+// OGCG64-LABEL: @test_builtin_xperm8_64(
+// OGCG64: call i64 @llvm.riscv.xperm8.i64(i64 {{%.*}}, i64 {{%.*}})
+// OGCG64: ret i64
+unsigned long long test_builtin_xperm8_64(unsigned long long a,
+                                          unsigned long long b) {
+  return __builtin_riscv_xperm8_64(a, b);
+}
+#endif

``````````

</details>


https://github.com/llvm/llvm-project/pull/195211