atomic_fix

Author: LeiWang1999

src/transform/atomicadd_vectorize.cc

@@ -232,15 +232,23 @@ class AtomicAddVectorizeRewriter : public StmtExprMutator {
       const IntImm memory_order =
           node->args.size() >= 3 ? Downcast<IntImm>(node->args[2]) : IntImm(0);
       Array<PrimExpr> new_args;
+      Call address_of_dst =
+          Call(DataType::Handle(), builtin::address_of(), {dst_node});
+      Call address_of_value =
+          Call(DataType::Handle(), builtin::address_of(), {value_node});
       if (vector_size_ == 4) {
         new_args.push_back(StringImm("AtomicAddx4"));
+        new_args.push_back(address_of_dst);
+        new_args.push_back(address_of_value);
       } else if (vector_size_ == 2) {
         new_args.push_back(StringImm("AtomicAddx2"));
+        new_args.push_back(address_of_dst);
+        new_args.push_back(address_of_value);
       } else {
         new_args.push_back(StringImm("AtomicAdd"));
+        new_args.push_back(dst_node);
+        new_args.push_back(value_node);
       }
-      new_args.push_back(dst_node);
-      new_args.push_back(value_node);
       new_args.push_back(memory_order);
 
       Call new_call =