Merge pull request #11 from JuliaRandom/remove-deps

Remove `deps` folder and check AES-NI in `__init__()`.

Author: sunoru

.github/workflows/build.yml

@@ -15,9 +15,12 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        julia-version: ["1.5", "1.6", nightly]
-        julia-arch: [x64]
+        julia-version: ["1.5", "1.7", nightly]
+        julia-arch: [x64, x86]
         os: [ubuntu-latest, windows-latest, macOS-latest]
+        exclude:
+          - os: macOS-latest
+            julia-arch: x86
 
     steps:
       - uses: actions/checkout@v2

deps/build.jl

@@ -18,7 +18,7 @@ module Random123
 
 using RandomNumbers
 
-export R123_USE_AESNI, set_counter!
+export set_counter!
 include("common.jl")
 
 export Threefry2x, Threefry4x
@@ -27,15 +27,38 @@ include("threefry.jl")
 export Philox2x, Philox4x
 include("philox.jl")
 
-if R123_USE_AESNI
-    include("aesni_common.jl")
-    export AESNI1x, AESNI4x
-    include("aesni.jl")
+"True when AES-NI has been enabled."
+const R123_USE_AESNI = Ref(false)
+export R123_USE_AESNI
+export AESNI1x, AESNI4x
+export ARS1x, ARS4x
+include("./aesni/module.jl")
 
-    export ARS1x, ARS4x
-    include("ars.jl")
-else
-    @warn "AES-NI is not enabled, so AESNI and ARS are not available."
+function __init__()
+
+    R123_USE_AESNI[] = try
+        cmd = Base.julia_cmd()
+        push!(
+            cmd.exec, "-e",
+            "const __m128i = NTuple{2, VecElement{UInt64}};" *
+            "@assert ccall(\"llvm.x86.aesni.aeskeygenassist\", " *
+            "llvmcall, __m128i, (__m128i, UInt8), " *
+            "__m128i((0x0123456789123450, 0x9876543210987654)), 0x1) ≡ " *
+            "__m128i((0x857c266f7c266e85, 0x2346382146382023))"
+        )
+        success(cmd)
+    catch e
+        @show e
+        false
+    end
+    
+    if R123_USE_AESNI[]
+        @eval using ._AESNIModule
+    else
+        @warn "AES-NI is not enabled, so AESNI and ARS are not available."
+    end
+
+    nothing
 end
 
 end

deps/has_aesni.jl

@@ -1,6 +1,9 @@
 using Base: llvmcall
 import Base.(+)
 
+using ..Random123: R123Generator1x, R123Generator4x
+import ..Random123: random123_r, set_counter!
+
 const __m128i = NTuple{2, VecElement{UInt64}}
 Base.convert(::Type{__m128i}, x::UInt128) = unsafe_load(Ptr{__m128i}(pointer_from_objref(Ref(x))))
 Base.convert(::Type{UInt128}, x::__m128i) = unsafe_load(Ptr{UInt128}(pointer_from_objref(Ref(x))))

src/Random123.jl

@@ -0,0 +1,14 @@
+module _AESNIModule
+
+using ..Random123
+
+export __m128i, AESNIKey
+include("common.jl")
+
+export AESNI1x, AESNI4x
+include("aesni.jl")
+
+export ARS1x, ARS4x
+include("ars.jl")
+
+end

src/aesni.jl renamed to src/aesni/aesni.jl

@@ -2,12 +2,6 @@ import Libdl
 import Random: rand, seed!
 import RandomNumbers: AbstractRNG
 
-const _dep_dir = joinpath(dirname(@__FILE__), "../deps/")
-include_dependency(joinpath(_dep_dir, "build.log"))
-
-"True when AES-NI has been enabled."
-const R123_USE_AESNI = isfile(joinpath(_dep_dir, "aes-ni"))
-
 const R123Array1x{T<:Union{UInt128}}        = NTuple{1, T}
 const R123Array2x{T<:Union{UInt32, UInt64}} = NTuple{2, T}
 const R123Array4x{T<:Union{UInt32, UInt64}} = NTuple{4, T}

src/ars.jl renamed to src/aesni/ars.jl

@@ -1,4 +1,4 @@
-if R123_USE_AESNI
+if R123_USE_AESNI[]
 
 import Random: seed!
 using Test: @test