llvm-reduce: Add atomic syncscope reduction

[arsenm]

b1e17199052a757e0173e9570e8ce52eec2b0048

llvm/test/tools/llvm-reduce/reduce-atomic-syncscope.ll

+; RUN: llvm-reduce --abort-on-invalid-reduction --delta-passes=syncscopes --test FileCheck --test-arg --check-prefixes=INTERESTING,CHECK --test-arg %s --test-arg --input-file %s -o %t
+; RUN: FileCheck -check-prefixes=RESULT,CHECK %s < %t
+
+; CHECK-LABEL: @load_syncscope_keep(
+; INTERESTING: syncscope("agent")
+; RESULT: %op = load atomic i32, ptr %ptr syncscope("agent") seq_cst, align 4
+define i32 @load_syncscope_keep(ptr %ptr) {
+  %op = load atomic i32, ptr %ptr syncscope("agent") seq_cst, align 4
+  ret i32 %op
+}
+
+; CHECK-LABEL: @load_syncscope_drop(
+; INTERESTING: load atomic
+; RESULT: %op = load atomic i32, ptr %ptr seq_cst, align 4
+define i32 @load_syncscope_drop(ptr %ptr) {
+  %op = load atomic i32, ptr %ptr syncscope("agent") seq_cst, align 4
+  ret i32 %op
+}
+
+; CHECK-LABEL: @store_syncscope_keep(
+; INTERESTING: syncscope("agent")
+; RESULT: store atomic i32 0, ptr %ptr syncscope("agent") seq_cst, align 4
+define void @store_syncscope_keep(ptr %ptr) {
+  store atomic i32 0, ptr %ptr syncscope("agent") seq_cst, align 4
+  ret void
+}
+
+; CHECK-LABEL: @store_syncscope_drop(
+; INTERESTING: store
+; RESULT: store atomic i32 0, ptr %ptr seq_cst, align 4
+define void @store_syncscope_drop(ptr %ptr) {
+  store atomic i32 0, ptr %ptr syncscope("agent") seq_cst, align 4
+  ret void
+}
+
+; CHECK-LABEL: @atomicrmw_syncscope_keep(
+; INTERESTING: syncscope("agent")
+; RESULT: %val = atomicrmw add ptr %ptr, i32 3 syncscope("agent") seq_cst, align 4
+define i32 @atomicrmw_syncscope_keep(ptr %ptr) {
+  %val = atomicrmw add ptr %ptr, i32 3 syncscope("agent") seq_cst, align 4
+  ret i32 %val
+}
+
+; CHECK-LABEL: @atomicrmw_syncscope_drop(
+; INTERESTING: atomicrmw
+; RESULT: %val = atomicrmw add ptr %ptr, i32 3 seq_cst, align 4
+define i32 @atomicrmw_syncscope_drop(ptr %ptr) {
+  %val = atomicrmw add ptr %ptr, i32 3 seq_cst
+  ret i32 %val
+}
+
+; CHECK-LABEL: @cmpxchg_syncscope_keep(
+; INTERESTING: syncscope("agent")
+; RESULT: %val = cmpxchg ptr %ptr, i32 %old, i32 %in syncscope("agent") seq_cst seq_cst, align 4
+define { i32, i1 } @cmpxchg_syncscope_keep(ptr %ptr, i32 %old, i32 %in) {
+  %val = cmpxchg ptr %ptr, i32 %old, i32 %in syncscope("agent") seq_cst seq_cst
+  ret { i32, i1 } %val
+}
+
+; CHECK-LABEL: @cmpxchg_syncscope_drop(
+; INTERESTING: = cmpxchg ptr
+; RESULT: %val = cmpxchg ptr %ptr, i32 %old, i32 %in seq_cst seq_cst, align 4
+define { i32, i1 } @cmpxchg_syncscope_drop(ptr %ptr, i32 %old, i32 %in) {
+  %val = cmpxchg ptr %ptr, i32 %old, i32 %in syncscope("agent") seq_cst seq_cst
+  ret { i32, i1 } %val
+}
+
+; CHECK-LABEL: @fence_syncscope_keep(
+; INTERESTING: syncscope("agent")
+; RESULT: fence syncscope("agent") acquire
+define void @fence_syncscope_keep() {
+  fence syncscope("agent") acquire
+  ret void
+}
+
+; CHECK-LABEL: @fence_syncscope_drop(
+; INTERESTING: fence
+; RESULT: fence acquire
+define void @fence_syncscope_drop() {
+  fence syncscope("agent") acquire
+  ret void
+}

llvm/tools/llvm-reduce/DeltaManager.cpp

do {                                                                         \
     DELTA_PASS("operands-nan", reduceOperandsNaNDeltaPass)                     \
     DELTA_PASS("operands-to-args", reduceOperandsToArgsDeltaPass)              \
     DELTA_PASS("operands-skip", reduceOperandsSkipDeltaPass)                   \
     DELTA_PASS("operand-bundles", reduceOperandBundesDeltaPass)                \
     DELTA_PASS("attributes", reduceAttributesDeltaPass)                        \
     DELTA_PASS("module-data", reduceModuleDataDeltaPass)                       \
     DELTA_PASS("opcodes", reduceOpcodesDeltaPass)                              \
     DELTA_PASS("volatile", reduceVolatileInstructionsDeltaPass)                \
+    DELTA_PASS("syncscopes", reduceAtomicSyncScopesDeltaPass)                  \
     DELTA_PASS("instruction-flags", reduceInstructionFlagsDeltaPass)           \
 } while (false)
 
 #define DELTA_PASSES_MIR                                                       \
   do {                                                                         \
     DELTA_PASS("instructions", reduceInstructionsMIRDeltaPass)                 \
     DELTA_PASS("ir-instruction-references",                                    \
                reduceIRInstructionReferencesDeltaPass)                         \

llvm/tools/llvm-reduce/deltas/ReduceMemoryOperations.h

 //===- ReduceMemoryOperations.h - Specialized Delta Pass --------*- C++ -*-===//
 //
 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 // See https://llvm.org/LICENSE.txt for license information.
 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 //
 //===----------------------------------------------------------------------===//
 
 #ifndef LLVM_TOOLS_LLVM_REDUCE_DELTAS_REDUCEMEMORYOPERATIONS_H
 #define LLVM_TOOLS_LLVM_REDUCE_DELTAS_REDUCEMEMORYOPERATIONS_H
 
 #include "TestRunner.h"
 
 namespace llvm {
 void reduceVolatileInstructionsDeltaPass(TestRunner &Test);
+void reduceAtomicSyncScopesDeltaPass(TestRunner &Test);
 } // namespace llvm
 
 #endif

llvm/tools/llvm-reduce/deltas/ReduceMemoryOperations.cpp

 static void removeVolatileInModule(Oracle &O, Module &Mod) {
   for (Function &F : Mod)
     removeVolatileInFunction(O, F);
 }
 
 void llvm::reduceVolatileInstructionsDeltaPass(TestRunner &Test) {
   runDeltaPass(Test, removeVolatileInModule, "Reducing Volatile Instructions");
 }
+
+static void reduceAtomicSyncScopesInFunction(Oracle &O, Function &F) {
+  for (Instruction &I : instructions(F)) {
+    if (LoadInst *LI = dyn_cast<LoadInst>(&I)) {
+      if (LI->getSyncScopeID() != SyncScope::System && !O.shouldKeep())
+        LI->setSyncScopeID(SyncScope::System);
+    } else if (StoreInst *SI = dyn_cast<StoreInst>(&I)) {
+      if (SI->getSyncScopeID() != SyncScope::System && !O.shouldKeep())
+        SI->setSyncScopeID(SyncScope::System);
+    } else if (AtomicRMWInst *RMW = dyn_cast<AtomicRMWInst>(&I)) {
+      if (RMW->getSyncScopeID() != SyncScope::System && !O.shouldKeep())
+        RMW->setSyncScopeID(SyncScope::System);
+    } else if (AtomicCmpXchgInst *CmpXChg = dyn_cast<AtomicCmpXchgInst>(&I)) {
+      if (CmpXChg->getSyncScopeID() != SyncScope::System && !O.shouldKeep())
+        CmpXChg->setSyncScopeID(SyncScope::System);
+    } else if (FenceInst *Fence = dyn_cast<FenceInst>(&I)) {
+      if (Fence->getSyncScopeID() != SyncScope::System && !O.shouldKeep())
+        Fence->setSyncScopeID(SyncScope::System);
+    }
+  }
+}
+
+static void reduceAtomicSyncScopesInModule(Oracle &O, Module &Mod) {
+  for (Function &F : Mod)
+    reduceAtomicSyncScopesInFunction(O, F);
+}
+
+void llvm::reduceAtomicSyncScopesDeltaPass(TestRunner &Test) {
+  runDeltaPass(Test, reduceAtomicSyncScopesInModule,
+               "Reducing Atomic Sync Scopes");
+}