[CIR][CIRGen][Builtin][Neon] Lower vrnd64z and vrnd64zq

Author: AmrDeveloper

clang/lib/CIR/CodeGen/CIRGenBuiltinAArch64.cpp

@@ -2627,6 +2627,14 @@ mlir::Value CIRGenFunction::emitCommonNeonBuiltinExpr(
     argTypes.push_back(vTy);
     break;
   }
+  case NEON::BI__builtin_neon_vrnd64z_f32:
+  case NEON::BI__builtin_neon_vrnd64zq_f32:
+  case NEON::BI__builtin_neon_vrnd64z_f64:
+  case NEON::BI__builtin_neon_vrnd64zq_f64: {
+    intrincsName = "aarch64.neon.frint64z";
+    argTypes.push_back(vTy);
+    break;
+  }
   case NEON::BI__builtin_neon_vshl_v:
   case NEON::BI__builtin_neon_vshlq_v: {
     return builder.create<cir::ShiftOp>(
@@ -4206,12 +4214,6 @@ CIRGenFunction::emitAArch64BuiltinExpr(unsigned BuiltinID, const CallExpr *E,
   case NEON::BI__builtin_neon_vrnd64xq_f64: {
     llvm_unreachable("NEON::BI__builtin_neon_vrnd64xq_f64 NYI");
   }
-  case NEON::BI__builtin_neon_vrnd64z_f32:
-  case NEON::BI__builtin_neon_vrnd64zq_f32:
-  case NEON::BI__builtin_neon_vrnd64z_f64:
-  case NEON::BI__builtin_neon_vrnd64zq_f64: {
-    llvm_unreachable("NEON::BI__builtin_neon_vrnd64zq_f64 NYI");
-  }
   case NEON::BI__builtin_neon_vrnd_v:
   case NEON::BI__builtin_neon_vrndq_v: {
     llvm_unreachable("NEON::BI__builtin_neon_vrndq_v NYI");

clang/test/CIR/CodeGen/AArch64/v8.5a-neon-frint3264-intrinsic.c

@@ -73,16 +73,30 @@ float32x4_t test_vrnd32xq_f32(float32x4_t a) {
 // CHECK-LABEL: test_vrnd64z_f32
 // CHECK:  [[RND:%.*]] =  call <2 x float> @llvm.aarch64.neon.frint64z.v2f32(<2 x float> %a)
 // CHECK:  ret <2 x float> [[RND]]
-// float32x2_t test_vrnd64z_f32(float32x2_t a) {
-//   return vrnd64z_f32(a);
-// }
+float32x2_t test_vrnd64z_f32(float32x2_t a) {
+  return vrnd64z_f32(a);
+
+  // CIR-LABEL: vrnd64z_f32
+  // CIR: [[TMP0:%.*]] = cir.llvm.intrinsic "aarch64.neon.frint64z" {{.*}} : (!cir.vector<!cir.float x 2>) -> !cir.vector<!cir.float x 2>
+
+  // LLVM-LABEL: @test_vrnd64z_f32
+  // LLVM:  [[RND:%.*]] =  call <2 x float> @llvm.aarch64.neon.frint64z.v2f32(<2 x float> %0)
+  // LLVM:  ret <2 x float> [[RND]]
+}
 
 // CHECK-LABEL: test_vrnd64zq_f32
 // CHECK:  [[RND:%.*]] =  call <4 x float> @llvm.aarch64.neon.frint64z.v4f32(<4 x float> %a)
 // CHECK:  ret <4 x float> [[RND]]
-// float32x4_t test_vrnd64zq_f32(float32x4_t a) {
-//   return vrnd64zq_f32(a);
-// }
+float32x4_t test_vrnd64zq_f32(float32x4_t a) {
+  return vrnd64zq_f32(a);
+
+  // CIR-LABEL: vrnd64zq_f32
+  // CIR: [[TMP0:%.*]] = cir.llvm.intrinsic "aarch64.neon.frint64z" {{.*}} : (!cir.vector<!cir.float x 4>) -> !cir.vector<!cir.float x 4>
+
+  // LLVM-LABEL: @test_vrnd64zq_f32
+  // LLVM:  [[RND:%.*]] =  call <4 x float> @llvm.aarch64.neon.frint64z.v4f32(<4 x float> %0)
+  // LLVM:  ret <4 x float> [[RND]]
+}
 
 float64x1_t test_vrnd32x_f64(float64x1_t a) {
   return vrnd32x_f64(a);
@@ -138,13 +152,27 @@ float64x2_t test_vrnd32xq_f64(float64x2_t a) {
 // CHECK-LABEL: test_vrnd64z_f64
 // CHECK:  [[RND:%.*]] =  call <1 x double> @llvm.aarch64.neon.frint64z.v1f64(<1 x double> %a)
 // CHECK:  ret <1 x double> [[RND]]
-// float64x1_t test_vrnd64z_f64(float64x1_t a) {
-//   return vrnd64z_f64(a);
-// }
+float64x1_t test_vrnd64z_f64(float64x1_t a) {
+  return vrnd64z_f64(a);
+
+  // CIR-LABEL: vrnd64z_f64
+  // CIR: [[TMP0:%.*]] = cir.llvm.intrinsic "aarch64.neon.frint64z" {{.*}} : (!cir.vector<!cir.double x 1>) -> !cir.vector<!cir.double x 1>
+
+  // LLVM-LABEL: @test_vrnd64z_f64
+  // LLVM:  [[RND:%.*]] =  call <1 x double> @llvm.aarch64.neon.frint64z.v1f64(<1 x double> %0)
+  // LLVM:  ret <1 x double> [[RND]]
+}
 
 // CHECK-LABEL: test_vrnd64zq_f64
 // CHECK:  [[RND:%.*]] =  call <2 x double> @llvm.aarch64.neon.frint64z.v2f64(<2 x double> %a)
 // CHECK:  ret <2 x double> [[RND]]
-// float64x2_t test_vrnd64zq_f64(float64x2_t a) {
-//   return vrnd64zq_f64(a);
-// }
+float64x2_t test_vrnd64zq_f64(float64x2_t a) {
+  return vrnd64zq_f64(a);
+
+  // CIR-LABEL: vrnd64zq_f64
+  // CIR: [[TMP0:%.*]] = cir.llvm.intrinsic "aarch64.neon.frint64z" {{.*}} : (!cir.vector<!cir.double x 2>) -> !cir.vector<!cir.double x 2>
+
+  // LLVM-LABEL: @test_vrnd64zq_f64
+  // LLVM:  [[RND:%.*]] =  call <2 x double> @llvm.aarch64.neon.frint64z.v2f64(<2 x double> %0)
+  // LLVM:  ret <2 x double> [[RND]]
+}