Skip to content

Commit fbf59b5

Browse files
committed
skip rewrite of analyze_binary
avoids issue with `mapreduce`
1 parent 8b98e66 commit fbf59b5

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

Project.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
4848
MPI = "da04e1cc-30fd-572f-bb4f-1f8673147195"
4949
NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
5050
NPZ = "15e1cf62-19b3-5cfa-8e77-841668bca605"
51+
OMEinsum = "ebe7aa44-baf0-506c-a96f-8464559b3922"
5152
OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
5253
OneHotArrays = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f"
5354
PythonCall = "6099a3de-0909-46bc-b1f4-468b9a2dfc0d"
@@ -73,6 +74,7 @@ ReactantKernelAbstractionsExt = "KernelAbstractions"
7374
ReactantMPIExt = "MPI"
7475
ReactantNNlibExt = ["NNlib", "Statistics"]
7576
ReactantNPZExt = "NPZ"
77+
ReactantOMEinsumExt = "OMEinsum"
7678
ReactantOffsetArraysExt = "OffsetArrays"
7779
ReactantOneHotArraysExt = "OneHotArrays"
7880
ReactantPythonCallExt = "PythonCall"
@@ -111,6 +113,7 @@ LinearAlgebra = "1.10"
111113
MPI = "0.20"
112114
NNlib = "0.9.26"
113115
NPZ = "0.4"
116+
OMEinsum = "0.9"
114117
OffsetArrays = "1"
115118
OneHotArrays = "0.2.10"
116119
OrderedCollections = "1.1"

ext/ReactantOMEinsumExt.jl

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
module ReactantOMEinsumExt
2+
3+
using Reactant: @skip_rewrite_func
4+
using OMEinsum
5+
6+
function __init__()
7+
@skip_rewrite_func OMEinsum.analyze_binary
8+
end
9+
10+
end

0 commit comments

Comments
 (0)