Remove Cassette from Reactant extension

bsc-quantic · Jul 16, 2024 · 6bed39a · 6bed39a
1 parent 55d9b49
commit 6bed39a
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/ext/MuscleReactantExt.jl b/ext/MuscleReactantExt.jl
@@ -4,7 +4,6 @@ using Muscle
 using Reactant
 const MLIR = Reactant.MLIR
 const stablehlo = MLIR.Dialects.stablehlo
-const Cassette = Reactant.Cassette
 
 function Muscle.einsum(ic, @nospecialize(a::Reactant.TracedRArray{Ta,Sa,Na}), ia, @nospecialize(b::Reactant.TracedRArray{Tb,Sb,Nb}), ib) where {Ta,Sa,Na,Tb,Sb,Nb}
     T = promote_type(Ta, Tb)
@@ -33,6 +32,4 @@ function Muscle.einsum(ic, @nospecialize(a::Reactant.TracedRArray{T,S,N}), ia) w
     return Reactant.TracedRArray{T,rsize,length(ic)}((), result)
 end
 
-Cassette.overdub(ctx::Reactant.TraceCtx, f::typeof(Muscle.einsum), args...; kwargs...) = f(args...; kwargs...)
-
 end