implement syncscopes everywhere

Author: vchuravy

src/UnsafeAtomics.jl

@@ -1,6 +1,7 @@
 baremodule UnsafeAtomics
 
 abstract type Ordering end
+abstract type SyncScope end
 
 function load end
 function store! end
@@ -30,6 +31,7 @@ using ..UnsafeAtomics: UnsafeAtomics, Ordering, right
 
 include("utils.jl")
 include("orderings.jl")
+include("syncscopes.jl")
 include("core.jl")
 
 end  # module Internal
@@ -45,4 +47,8 @@ const seq_cst = Internal.seq_cst
 const acquire_release = acq_rel
 const sequentially_consistent = seq_cst
 
+# SyncScope
+const none = Internal.none
+const singlethread = Internal.singlethread
+
 end  # baremodule UnsafeAtomics

src/core.jl

@@ -4,6 +4,12 @@
 @inline UnsafeAtomics.modify!(ptr, op, x) = UnsafeAtomics.modify!(ptr, op, x, seq_cst)
 @inline UnsafeAtomics.fence() = UnsafeAtomics.fence(seq_cst)
 
+@inline UnsafeAtomics.load(x, ord) = UnsafeAtomics.load(x, ord, none)
+@inline UnsafeAtomics.store!(x, v, ord) = UnsafeAtomics.store!(x, v, ord, none)
+@inline UnsafeAtomics.cas!(x, cmp, new, ord) = UnsafeAtomics.cas!(x, cmp, new, ord, ord, none)
+@inline UnsafeAtomics.modify!(ptr, op, x, ord) = UnsafeAtomics.modify!(ptr, op, x, ord, none)
+@inline UnsafeAtomics.fence(ord) = UnsafeAtomics.fence(ord., none)
+
 #! format: off
 # https://github.com/JuliaLang/julia/blob/v1.6.3/base/atomics.jl#L23-L30
 if Sys.ARCH == :i686 || startswith(string(Sys.ARCH), "arm") ||
@@ -45,8 +51,9 @@ const OP_RMW_TABLE = [
 for (op, rmwop) in OP_RMW_TABLE
     fn = Symbol(rmwop, "!")
     @eval @inline UnsafeAtomics.$fn(x, v) = UnsafeAtomics.$fn(x, v, seq_cst)
-    @eval @inline UnsafeAtomics.$fn(ptr, x, ord) =
-        first(UnsafeAtomics.modify!(ptr, $op, x, ord))
+    @eval @inline UnsafeAtomics.$fn(x, v, ord) = UnsafeAtomics.$fn(x, v, ord, none) 
+    @eval @inline UnsafeAtomics.$fn(ptr, x, ord, scope) =
+        first(UnsafeAtomics.modify!(ptr, $op, x, ord, scope))
 end
 
 const ATOMIC_INTRINSICS = isdefined(Core.Intrinsics, :atomic_pointerref)

src/syncscopes.jl

@@ -0,0 +1,9 @@
+struct LLVMSyncScope{name} <: SyncScope end
+
+const none = LLVMSyncScope{Symbol("")}()
+const singlethread = LLVMSyncScope{:singlethread}()
+
+llvm_syncscope(::LLVMSyncScope{name}) where {name} = name
+
+Base.string(s::LLVMSyncScope) = String(llvm_syncscope(s))
+Base.print(io::IO, s::LLVMSyncScope) = print(io, string(s))