support julia v1.8

TensorBFS · Aug 16, 2024 · 0d3d1ba · 0d3d1ba
1 parent 88ef130
commit 0d3d1ba
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 34 deletions.
diff --git a/Project.toml b/Project.toml
@@ -26,11 +26,12 @@ KaHyPar = "0.3"
 StatsBase = "0.34"
 Suppressor = "0.2"
 LuxorGraphPlot = "0.5.1"
-TreeWidthSolver = "0.3"
-julia = "1.9"
+TreeWidthSolver = "0.3.1"
+julia = "1.8"
 
 [extras]
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
+KaHyPar = "2a6221f6-aa48-11e9-3542-2d9e0ef01880"
 OMEinsum = "ebe7aa44-baf0-506c-a96f-8464559b3922"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

diff --git a/test/interfaces.jl b/test/interfaces.jl
@@ -15,7 +15,17 @@ using OMEinsum
     xs = [[randn(2,2) for i=1:150]..., [randn(2) for i=1:100]...]
 
     results = Float64[]
-    for optimizer in [TreeSA(ntrials=1), TreeSA(ntrials=1, nslices=5), GreedyMethod(), KaHyParBipartite(sc_target=18), SABipartite(sc_target=18, ntrials=1)]
+    for optimizer in [TreeSA(ntrials=1), TreeSA(ntrials=1, nslices=5), GreedyMethod(), SABipartite(sc_target=18, ntrials=1)]
+        for simplifier in (nothing, MergeVectors(), MergeGreedy())
+            @info "optimizer = $(optimizer), simplifier = $(simplifier)"
+            res = optimize_code(code,uniformsize(code, 2), optimizer, simplifier)
+            tc, sc = OMEinsum.timespace_complexity(res, uniformsize(code, 2))
+            @test sc <= 18
+            push!(results, res(xs...)[])
+        end
+    end
+    if isdefined(Base, :get_extension)
+        optimizer = KaHyParBipartite(sc_target=18)
         for simplifier in (nothing, MergeVectors(), MergeGreedy())
             @info "optimizer = $(optimizer), simplifier = $(simplifier)"
             res = optimize_code(code,uniformsize(code, 2), optimizer, simplifier)
@@ -32,7 +42,16 @@ using OMEinsum
     xs = [[randn(2,2) for i=1:15]..., [randn(2) for i=1:10]...]
 
     results = Float64[]
-    for optimizer in [TreeSA(ntrials=1), TreeSA(ntrials=1, nslices=5), GreedyMethod(), KaHyParBipartite(sc_target=18), SABipartite(sc_target=18, ntrials=1), ExactTreewidth()]
+    for optimizer in [TreeSA(ntrials=1), TreeSA(ntrials=1, nslices=5), GreedyMethod(), SABipartite(sc_target=18, ntrials=1), ExactTreewidth()]
+        for simplifier in (nothing, MergeVectors(), MergeGreedy())
+            @info "optimizer = $(optimizer), simplifier = $(simplifier)"
+            res = optimize_code(small_code,uniformsize(small_code, 2), optimizer, simplifier)
+            tc, sc = OMEinsum.timespace_complexity(res, uniformsize(small_code, 2))
+            push!(results, res(xs...)[])
+        end
+    end
+    if isdefined(Base, :get_extension)
+        optimizer = KaHyParBipartite(sc_target=18)
         for simplifier in (nothing, MergeVectors(), MergeGreedy())
             @info "optimizer = $(optimizer), simplifier = $(simplifier)"
             res = optimize_code(small_code,uniformsize(small_code, 2), optimizer, simplifier)
@@ -53,7 +72,7 @@ end
     @test optimize_code(code, sizes, GreedyMethod()) == sne
     @test optimize_code(code, sizes, TreeSA()) == SlicedEinsum(Char[], dne)
     @test optimize_code(code, sizes, TreeSA(nslices=2)) == SlicedEinsum(Char[], dne)
-    @test optimize_code(code, sizes, KaHyParBipartite(sc_target=25)) == dne
+    isdefined(Base, :get_extension) &&  (@test optimize_code(code, sizes, KaHyParBipartite(sc_target=25)) == dne)
     @test optimize_code(code, sizes, SABipartite(sc_target=25)) == dne
 end
 
@@ -79,32 +98,33 @@ end
     @test 10 * 2^sc2 > pm2 > 2^sc2
 end
 
+if isdefined(Base, :get_extension)
+    @testset "kahypar regression test" begin
+        code = ein"i->"
+        optcode = optimize_code(code, Dict('i'=>4), KaHyParBipartite(; sc_target=10, max_group_size=10))
+        @test optcode isa NestedEinsum
+        x = randn(4)
+        @test optcode(x) ≈ code(x)
 
-@testset "kahypar regression test" begin
-    code = ein"i->"
-    optcode = optimize_code(code, Dict('i'=>4), KaHyParBipartite(; sc_target=10, max_group_size=10))
-    @test optcode isa NestedEinsum
-    x = randn(4)
-    @test optcode(x) ≈ code(x)
-
-    code = ein"i,j->"
-    optcode = optimize_code(code, Dict('i'=>4, 'j'=>4), KaHyParBipartite(; sc_target=10, max_group_size=10))
-    @test optcode isa NestedEinsum
-    x = randn(4)
-    y = randn(4)
-    @test optcode(x, y) ≈ code(x, y)
+        code = ein"i,j->"
+        optcode = optimize_code(code, Dict('i'=>4, 'j'=>4), KaHyParBipartite(; sc_target=10, max_group_size=10))
+        @test optcode isa NestedEinsum
+        x = randn(4)
+        y = randn(4)
+        @test optcode(x, y) ≈ code(x, y)
 
-    code = ein"ij,jk,kl->ijl"
-    println(code)
-    optcode = optimize_code(code, Dict('i'=>4, 'j'=>4, 'k'=>4, 'l'=>4), KaHyParBipartite(; sc_target=4, max_group_size=2))
-    println(optcode)
-    @test optcode isa NestedEinsum
-    a, b, c = [rand(4,4) for i=1:4]
-    @test optcode(a, b, c) ≈ code(a, b, c)
+        code = ein"ij,jk,kl->ijl"
+        println(code)
+        optcode = optimize_code(code, Dict('i'=>4, 'j'=>4, 'k'=>4, 'l'=>4), KaHyParBipartite(; sc_target=4, max_group_size=2))
+        println(optcode)
+        @test optcode isa NestedEinsum
+        a, b, c = [rand(4,4) for i=1:4]
+        @test optcode(a, b, c) ≈ code(a, b, c)
 
-    code = ein"ij,jk,kl->ijl"
-    optcode = optimize_code(code, Dict('i'=>3, 'j'=>3, 'k'=>3, 'l'=>3), KaHyParBipartite(; sc_target=4, max_group_size=2))
-    @test optcode isa NestedEinsum
-    a, b, c = [rand(3,3) for i=1:4]
-    @test optcode(a, b, c) ≈ code(a, b, c)
+        code = ein"ij,jk,kl->ijl"
+        optcode = optimize_code(code, Dict('i'=>3, 'j'=>3, 'k'=>3, 'l'=>3), KaHyParBipartite(; sc_target=4, max_group_size=2))
+        @test optcode isa NestedEinsum
+        a, b, c = [rand(3,3) for i=1:4]
+        @test optcode(a, b, c) ≈ code(a, b, c)
+    end
 end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -13,8 +13,10 @@ end
     include("sa.jl")
 end
 
-@testset "kahypar" begin
-    include("kahypar.jl")
+if isdefined(Base, :get_extension)
+    @testset "kahypar" begin
+        include("kahypar.jl")
+    end
 end
 
 @testset "treesa" begin
@@ -38,6 +40,8 @@ end
 end
 
 # testing the extension `LuxorTensorPlot` for visualization
-@testset "visualization" begin
-    include("visualization.jl")
+if isdefined(Base, :get_extension)
+    @testset "visualization" begin
+        include("visualization.jl")
+    end
 end