dress-graph/examples/julia/mpi_omp.jl at main · velicast/dress-graph · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# mpi_omp.jl — Rook vs Shrikhande with Δ¹-DRESS (MPI + OpenMP)
# Keeps multisets and compares them to guarantee distinguishability.
#
# Run:
#   mpirun -np 4 julia mpi_omp.jl
using MPI
MPI.Init()

using DRESS.MPI.OMP: delta_fit

# Rook L₂(4) = K₄ □ K₄ — 16 vertices, 96 directed edges
rook_s = [0,1,0,4,0,2,0,8,0,3,0,12,1,5,1,2,1,9,1,3,1,13,2,6,2,10,2,3,2,14,3,7,3,11,3,15,4,5,4,6,4,8,4,7,4,12,5,6,5,9,5,7,5,13,6,10,6,7,6,14,7,11,7,15,8,9,8,10,8,11,8,12,9,10,9,11,9,13,10,11,10,14,11,15,12,13,12,14,12,15,13,14,13,15,14,15]
rook_t = [1,0,4,0,2,0,8,0,3,0,12,0,5,1,2,1,9,1,3,1,13,1,6,2,10,2,3,2,14,2,7,3,11,3,15,3,5,4,6,4,8,4,7,4,12,4,6,5,9,5,7,5,13,5,10,6,7,6,14,6,11,7,15,7,9,8,10,8,11,8,12,8,10,9,11,9,13,9,11,10,14,10,15,11,13,12,14,12,15,12,14,13,15,13,15,14]

# Shrikhande — 16 vertices, 96 directed edges
shri_s = [0,4,0,12,0,1,0,3,0,5,0,15,1,5,1,13,1,2,1,6,1,12,2,6,2,14,2,3,2,7,2,13,3,7,3,15,3,4,3,14,4,8,4,5,4,7,4,9,5,9,5,6,5,10,6,10,6,7,6,11,7,11,7,8,8,12,8,9,8,11,8,13,9,13,9,10,9,14,10,14,10,11,10,15,11,15,11,12,12,13,12,15,13,14,14,15]
shri_t = [4,0,12,0,1,0,3,0,5,0,15,0,5,1,13,1,2,1,6,1,12,1,6,2,14,2,3,2,7,2,13,2,7,3,15,3,4,3,14,3,8,4,5,4,7,4,9,4,9,5,6,5,10,5,10,6,7,6,11,6,11,7,8,7,12,8,9,8,11,8,13,8,13,9,10,9,14,9,14,10,11,10,15,10,15,11,12,11,13,12,15,12,14,13,15,14]

dr = delta_fit(16, rook_s, rook_t; k=1, keep_multisets=true)
ds = delta_fit(16, shri_s, shri_t; k=1, keep_multisets=true)

if MPI.Comm_rank(MPI.COMM_WORLD) == 0
    println("Rook:       $(length(dr.histogram)) histogram entries, $(dr.num_subgraphs) subgraphs")
    println("Shrikhande: $(length(ds.histogram)) histogram entries, $(ds.num_subgraphs) subgraphs")
    println("Histograms differ:  $(dr.histogram != ds.histogram)")

    # Canonicalize: sort each row, then sort rows
    function canonicalize(ms)
        s = sort(ms, dims=2)          # sort within each row
        s[sortperm(eachrow(s)), :]    # sort rows lexicographically
    end

    cr = canonicalize(dr.multisets)
    cs = canonicalize(ds.multisets)
    ms_same = size(cr) == size(cs) && all(
        (isnan(a) && isnan(b)) || a == b for (a, b) in zip(cr, cs)
    )
    println("Multisets differ:   $(!ms_same)")
end

MPI.Finalize()