Turn result to atomic

xcnick · xcnick · commit f72b7f4efd1f · 2020-06-24T10:57:12.000+08:00
diff --git a/aten/src/ATen/native/ReduceOps.cpp b/aten/src/ATen/native/ReduceOps.cpp
@@ -968,7 +968,7 @@ bool cpu_equal(const Tensor& self, const Tensor& other) {
   if (!self.is_same_size(other)) {
     return false;
   }
-  bool result = true;
+  std::atomic<bool> result{true};
   auto iter = TensorIteratorConfig()
     .add_input(self)
     .add_input(other)
@@ -993,7 +993,7 @@ bool cpu_equal(const Tensor& self, const Tensor& other) {
       }
     });
   });
-  return result;
+  return result.load();
 }
 
 }} // namespace at::native

Original file line number	Diff line number	Diff line change
`@@ -968,7 +968,7 @@ bool cpu_equal(const Tensor& self, const Tensor& other) {`
`968`	`968`	`if (!self.is_same_size(other)) {`
`969`	`969`	`return false;`
`970`	`970`	`}`
`971`		`- bool result = true;`
	`971`	`+ std::atomic<bool> result{true};`
`972`	`972`	`auto iter = TensorIteratorConfig()`
`973`	`973`	`.add_input(self)`
`974`	`974`	`.add_input(other)`
`@@ -993,7 +993,7 @@ bool cpu_equal(const Tensor& self, const Tensor& other) {`
`993`	`993`	`}`
`994`	`994`	`});`
`995`	`995`	`});`
`996`		`- return result;`
	`996`	`+ return result.load();`
`997`	`997`	`}`
`998`	`998`
`999`	`999`	`}} // namespace at::native`