Replace isAtomic field with MemoryOrder enum in Stores

src/binaryen-c.cpp

@@ -2698,12 +2698,13 @@ void BinaryenLoadSetPtr(BinaryenExpressionRef expr,
 bool BinaryenStoreIsAtomic(BinaryenExpressionRef expr) {
   auto* expression = (Expression*)expr;
   assert(expression->is<Store>());
-  return static_cast<Store*>(expression)->isAtomic;
+  return static_cast<Store*>(expression)->isAtomic();
 }
 void BinaryenStoreSetAtomic(BinaryenExpressionRef expr, bool isAtomic) {
   auto* expression = (Expression*)expr;
   assert(expression->is<Store>());
-  static_cast<Store*>(expression)->isAtomic = isAtomic != 0;
+  static_cast<Store*>(expression)->order =
+    isAtomic ? MemoryOrder::SeqCst : MemoryOrder::Unordered;
 }
 uint32_t BinaryenStoreGetBytes(BinaryenExpressionRef expr) {
   auto* expression = (Expression*)expr;

src/ir/cost.h

@@ -105,7 +105,7 @@ struct CostAnalyzer : public OverriddenVisitor<CostAnalyzer, CostType> {
   }
   CostType visitStore(Store* curr) {
     return 2 + visit(curr->ptr) + visit(curr->value) +
-           AtomicCost * curr->isAtomic;
+           AtomicCost * curr->isAtomic();
   }
   CostType visitAtomicRMW(AtomicRMW* curr) {
     return AtomicCost + visit(curr->ptr) + visit(curr->value);

src/ir/effects.h

@@ -603,7 +603,7 @@ class EffectAnalyzer {
     }
     void visitStore(Store* curr) {
       parent.writesMemory = true;
-      parent.isAtomic |= curr->isAtomic;
+      parent.isAtomic |= curr->isAtomic();
       parent.implicitTrap = true;
     }
     void visitAtomicRMW(AtomicRMW* curr) {

src/ir/properties.h

@@ -505,7 +505,7 @@ inline MemoryOrder getMemoryOrder(Expression* curr) {
     return load->order;
   }
   if (auto* store = curr->dynCast<Store>()) {
-    return store->isAtomic ? MemoryOrder::SeqCst : MemoryOrder::Unordered;
+    return store->order;
   }
   if (curr->is<AtomicRMW>() || curr->is<AtomicWait>() ||
       curr->is<AtomicNotify>() || curr->is<AtomicFence>()) {

src/passes/I64ToI32Lowering.cpp

@@ -458,7 +458,7 @@ struct I64ToI32Lowering : public WalkerPass<PostWalker<I64ToI32Lowering>> {
       return;
     }
     assert(curr->offset + 4 > curr->offset);
-    assert(!curr->isAtomic && "atomic store not implemented");
+    assert(!curr->isAtomic() && "atomic store not implemented");
     TempVar highBits = fetchOutParam(curr->value);
     uint8_t bytes = curr->bytes;
     curr->bytes = std::min(curr->bytes, uint8_t(4));

src/passes/OptimizeInstructions.cpp

@@ -1282,7 +1282,7 @@ struct OptimizeInstructions
         // instead of wrapping to 32, just store some of the bits in the i64
         curr->valueType = Type::i64;
         curr->value = unary->value;
-      } else if (!curr->isAtomic && Abstract::hasAnyReinterpret(unary->op) &&
+      } else if (!curr->isAtomic() && Abstract::hasAnyReinterpret(unary->op) &&
                  curr->bytes == curr->valueType.getByteSize()) {
         // f32.store(y, f32.reinterpret_i32(x))  =>  i32.store(y, x)
         // f64.store(y, f64.reinterpret_i64(x))  =>  i64.store(y, x)

src/passes/Print.cpp

@@ -583,7 +583,7 @@ struct PrintExpressionContents
   }
   void visitStore(Store* curr) {
     prepareColor(o) << forceConcrete(curr->valueType);
-    if (curr->isAtomic) {
+    if (curr->isAtomic()) {
       o << ".atomic";
     }
     o << ".store";

src/passes/SafeHeap.cpp

@@ -55,7 +55,7 @@ static Name getStoreName(Store* curr) {
   std::string ret = "SAFE_HEAP_STORE_";
   ret += curr->valueType.toString();
   ret += "_" + std::to_string(curr->bytes) + "_";
-  if (curr->isAtomic) {
+  if (curr->isAtomic()) {
     ret += "A";
   } else {
     ret += std::to_string(curr->align);
@@ -270,7 +270,8 @@ struct SafeHeap : public Pass {
             continue;
           }
           for (auto isAtomic : {true, false}) {
-            store.isAtomic = isAtomic;
+            store.order =
+              isAtomic ? MemoryOrder::SeqCst : MemoryOrder::Unordered;
             if (isAtomic &&
                 !isPossibleAtomicOperation(
                   align, bytes, module->memories[0]->shared, valueType)) {

src/tools/fuzzing/fuzzing.cpp

@@ -3358,7 +3358,7 @@ Expression* TranslateToFuzzReader::makeStore(Type type) {
   }
   // make it atomic
   wasm.memories[0]->shared = true;
-  store->isAtomic = true;
+  store->order = MemoryOrder::SeqCst;
   store->align = store->bytes;
   return store;
 }

src/wasm-builder.h

@@ -428,14 +428,14 @@ class Builder {
                    Type type,
                    Name memory) {
     auto* ret = wasm.allocator.alloc<Store>();
-    ret->isAtomic = false;
     ret->bytes = bytes;
     ret->offset = offset;
     ret->align = align;
     ret->ptr = ptr;
     ret->value = value;
     ret->valueType = type;
     ret->memory = memory;
+    ret->order = MemoryOrder::Unordered;
     ret->finalize();
     return ret;
   }
@@ -446,7 +446,7 @@ class Builder {
                          Type type,
                          Name memory) {
     Store* store = makeStore(bytes, offset, bytes, ptr, value, type, memory);
-    store->isAtomic = true;
+    store->order = MemoryOrder::SeqCst;
     return store;
   }
   AtomicRMW* makeAtomicRMW(AtomicRMWOp op,

src/wasm-delegations-fields.def

@@ -364,7 +364,7 @@ DELEGATE_FIELD_CHILD(Store, ptr)
 DELEGATE_FIELD_INT(Store, bytes)
 DELEGATE_FIELD_ADDRESS(Store, offset)
 DELEGATE_FIELD_ADDRESS(Store, align)
-DELEGATE_FIELD_INT(Store, isAtomic)
+DELEGATE_FIELD_INT(Store, order)
 DELEGATE_FIELD_TYPE(Store, valueType)
 DELEGATE_FIELD_NAME_KIND(Store, memory, ModuleItemKind::Memory)
 DELEGATE_FIELD_CASE_END(Store)

src/wasm-interpreter.h

@@ -3966,7 +3966,7 @@ class ModuleRunnerBase : public ExpressionRunner<SubType> {
     auto memorySize = info.instance->getMemorySize(info.name);
     auto addr =
       info.instance->getFinalAddress(curr, ptr.getSingleValue(), memorySize);
-    if (curr->isAtomic) {
+    if (curr->isAtomic()) {
       info.instance->checkAtomicAddress(addr, curr->bytes, memorySize);
     }
     info.interface()->store(curr, addr, value.getSingleValue(), info.name);
@@ -5136,7 +5136,7 @@ class ModuleRunnerBase : public ExpressionRunner<SubType> {
     Store store;
     store.bytes = bytes;
     store.align = bytes;
-    store.isAtomic = true; // understatement
+    store.order = MemoryOrder::SeqCst;
     store.ptr = &ptr;
     store.value = &value;
     store.valueType = value.type;

src/wasm.h

@@ -1012,11 +1012,13 @@ class Store : public SpecificExpression<Expression::StoreId> {
   uint8_t bytes;
   Address offset;
   Address align;
-  bool isAtomic;
   Expression* ptr;
   Expression* value;
   Type valueType;
   Name memory;
+  MemoryOrder order;
+
+  bool isAtomic() const { return order != MemoryOrder::Unordered; }
 
   void finalize();
 };

src/wasm/wasm-stack.cpp

@@ -365,7 +365,7 @@ void BinaryInstWriter::visitLoad(Load* curr) {
 }
 
 void BinaryInstWriter::visitStore(Store* curr) {
-  if (!curr->isAtomic) {
+  if (!curr->isAtomic()) {
     switch (curr->valueType.getBasic()) {
       case Type::i32: {
         switch (curr->bytes) {

src/wasm/wasm-validator.cpp

@@ -1128,7 +1128,7 @@ void FunctionValidator::visitLoad(Load* curr) {
 void FunctionValidator::visitStore(Store* curr) {
   auto* memory = getModule()->getMemoryOrNull(curr->memory);
   shouldBeTrue(!!memory, curr, "memory.store memory must exist");
-  if (curr->isAtomic) {
+  if (curr->isAtomic()) {
     shouldBeTrue(getModule()->features.hasAtomics(),
                  curr,
                  "Atomic operations require threads [--enable-threads]");
@@ -1145,7 +1145,7 @@ void FunctionValidator::visitStore(Store* curr) {
   validateMemBytes(curr->bytes, curr->valueType, curr);
   validateOffset(curr->offset, memory, curr);
   validateAlignment(
-    curr->align, curr->valueType, curr->bytes, curr->isAtomic, curr);
+    curr->align, curr->valueType, curr->bytes, curr->isAtomic(), curr);
   shouldBeEqualOrFirstIsUnreachable(
     curr->ptr->type,
     memory->addressType,
@@ -1157,7 +1157,7 @@ void FunctionValidator::visitStore(Store* curr) {
                   "store value type must not be none");
   shouldBeEqualOrFirstIsUnreachable(
     curr->value->type, curr->valueType, curr, "store value type must match");
-  if (curr->isAtomic) {
+  if (curr->isAtomic()) {
     shouldBeIntOrUnreachable(
       curr->valueType, curr, "atomic stores must be of integers");
   }

src/wasm2js.h

@@ -1562,7 +1562,7 @@ Ref Wasm2JSBuilder::processExpression(Expression* curr,
           Fatal() << "Unhandled type in store: " << curr->valueType;
         }
       }
-      if (curr->isAtomic) {
+      if (curr->isAtomic()) {
         Ref call = ValueBuilder::makeCall(
           ValueBuilder::makeDot(ValueBuilder::makeName(ATOMICS), STORE));
         ValueBuilder::appendToCall(call, ret[1]);