[OpenMP] Fix begin_end_declare_variant.cpp

Author: zahiraam

clang/test/OpenMP/spirv_variant_match.cpp

@@ -8,11 +8,13 @@
 
 // RUN: %clang_cc1 -verify -fopenmp -x c++ -triple x86_64-unknown-unknown -fopenmp-targets=spirv64-intel -emit-llvm-bc %s -o %t-host.bc -DTARGET_KIND
 // RUN: %clang_cc1 -verify -triple spirv64-intel -aux-triple x86_64-unknown-unknown -fopenmp -fopenmp-is-target-device \
-// RUN: -fopenmp-host-ir-file-path %t-host.bc -nogpulib %s -emit-llvm -DTARGET_KIND -o - | FileCheck %s
+// RUN: -fopenmp-host-ir-file-path %t-host.bc -nogpulib %s -emit-llvm -DTARGET_KIND -o - | FileCheck %s --check-prefix=TDK_GPU
 
 // RUN: %clang_cc1 -verify -fopenmp -x c++ -triple x86_64-unknown-unknown -fopenmp-targets=spirv64-intel -emit-llvm-bc %s -o %t-host.bc
 // RUN: %clang_cc1 -verify -triple spirv64-intel -aux-triple x86_64-unknown-unknown -fopenmp -fopenmp-is-target-device \
-// RUN: -fopenmp-host-ir-file-path %t-host.bc -nogpulib %s -emit-llvm -o - | FileCheck %s
+// RUN: -fopenmp-host-ir-file-path %t-host.bc -nogpulib %s -emit-llvm -o - \
+// RUN: | FileCheck %s --check-prefix=DK_GPU
+
 
 // expected-no-diagnostics
 
@@ -33,9 +35,11 @@ int foo() { return 1; }
 #pragma omp end declare variant
 #pragma omp end declare target
 
-// CHECK-DAG: define{{.*}} @{{"_Z[0-9]+foo\$ompvariant\$.*"}}()
+// CHECK-DAG: define{{.*}}  @_Z3foov()
 
-// CHECK-DAG: call spir_func noundef i32 @{{"_Z[0-9]+foo\$ompvariant\$.*"}}()
+// CHECK-DAG: call spir_func noundef i32 @_Z3foov()
+// TDK_GPU-DAG: call spir_func noundef i32 @"_Z25foo$ompvariant$S3$s8$Pgpuv"()
+// DK_GPU-DAG: call spir_func noundef i32 @"_Z25foo$ompvariant$S2$s6$Pgpuv"()
 
 int main() {
   int res;