Add LoopVectorization.jl to tests and benchmarks to speed up Tullio (#66)

adrhill · web-flow · commit c6c8db9db52d · 2022-05-25T16:30:41.000+02:00
diff --git a/Project.toml b/Project.toml
@@ -28,10 +28,11 @@ julia = "1.6"
 
 [extras]
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
+LoopVectorization = "bdcacae8-1622-11e9-2a5c-532679323890"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ReferenceTests = "324d217c-45ce-50fc-942e-d289b448e8cf"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["JLD2", "Random", "ReferenceTests", "Suppressor", "Test"]
+test = ["JLD2", "LoopVectorization", "Random", "ReferenceTests", "Suppressor", "Test"]
diff --git a/benchmark/Project.toml b/benchmark/Project.toml
@@ -2,4 +2,5 @@
 BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
 ExplainableAI = "4f1bc3e1-d60d-4ed0-9367-9bdff9846d3b"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
+LoopVectorization = "bdcacae8-1622-11e9-2a5c-532679323890"
 PkgBenchmark = "32113eaa-f34f-5b0d-bd6c-c81e245fc73d"
diff --git a/benchmark/benchmarks.jl b/benchmark/benchmarks.jl
@@ -1,4 +1,5 @@
 using BenchmarkTools
+using LoopVectorization
 using Flux
 using ExplainableAI
 import ExplainableAI: modify_layer, lrp!
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,3 +1,4 @@
+using LoopVectorization
 using ExplainableAI
 using Test
 using ReferenceTests
diff --git a/test/test_rules.jl b/test/test_rules.jl
@@ -1,3 +1,4 @@
+using LoopVectorization
 using ExplainableAI
 using ExplainableAI: modify_params
 import ExplainableAI: modify_layer, lrp!
diff --git a/test/test_vgg11.jl b/test/test_vgg11.jl
@@ -1,3 +1,4 @@
+using LoopVectorization
 using ExplainableAI
 using Flux
 using JLD2

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`using BenchmarkTools`
	`2`	`+using LoopVectorization`
`2`	`3`	`using Flux`
`3`	`4`	`using ExplainableAI`
`4`	`5`	`import ExplainableAI: modify_layer, lrp!`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+using LoopVectorization`
`1`	`2`	`using ExplainableAI`
`2`	`3`	`using Test`
`3`	`4`	`using ReferenceTests`