[MLIR][OpenMP] Add a constraint to the Threadprivate Op

[kiranchandramohan]

Summary

Add a constraint to ensure that the operand and result of the
threadprivate operation are the same.

[peixin]

Nice.

mlir/include/mlir/Dialect/OpenMP/OpenMPOps.td

let extraClassDeclaration = [{
     AtomicUpdateOp getAtomicUpdateOp();
   }];
 }
 
 //===----------------------------------------------------------------------===//
 // [5.1] 2.21.2 threadprivate Directive
 //===----------------------------------------------------------------------===//
 
-def ThreadprivateOp : OpenMP_Op<"threadprivate"> {
+def ThreadprivateOp : OpenMP_Op<"threadprivate",
+                                [AllTypesMatch<["sym_addr", "tls_addr"]>]> {
   let summary = "threadprivate directive";
   let description = [{
     The threadprivate directive specifies that variables are replicated, with
     each thread having its own copy.
 
     The current implementation uses the OpenMP runtime to provide thread-local
     storage (TLS). Using the TLS feature of the LLVM IR will be supported in
     future.

mlir/test/Dialect/OpenMP/invalid.mlir

func.func @taskloop(%lb: i32, %ub: i32, %step: i32) {
   %testi64 = "test.i64"() : () -> (i64)
   // expected-error @below {{the grainsize clause and num_tasks clause are mutually exclusive and may not appear on the same taskloop directive}}
   omp.taskloop grain_size(%testi64: i64) num_tasks(%testi64: i64)
   for (%i, %j) : i32 = (%lb, %ub) to (%ub, %lb) step (%step, %step) {
     omp.terminator
   }
   return
 }
+
+// -----
+
+func.func @omp_threadprivate() {
+  %1 = llvm.mlir.addressof @_QFsubEx : !llvm.ptr<i32>
+  // expected-error @below {{op failed to verify that all of {sym_addr, tls_addr} have same type}}
+  %2 = omp.threadprivate %1 : !llvm.ptr<i32> -> memref<i32>
+  return
+}
+
+llvm.mlir.global internal @_QFsubEx() : i32