From 5a430a63a14cefb6f503f4582e04434d1098acb0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sergio=20S=C3=A1nchez=20Ram=C3=ADrez?= Date: Sun, 6 Oct 2024 17:07:07 -0400 Subject: [PATCH] Try remove ambiguity on `contract` with `TracedRArray` --- ext/TenetReactantExt.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ext/TenetReactantExt.jl b/ext/TenetReactantExt.jl index f75a271a3..814857166 100644 --- a/ext/TenetReactantExt.jl +++ b/ext/TenetReactantExt.jl @@ -124,8 +124,8 @@ end end function Tenet.contract( - a::Tensor{Ta,Na,Aa}, b::Tensor{Tb,Nb,Ab}; dims=(∩(inds(a), inds(b))), out=nothing -) where {Ta,Na,Aa<:TracedRArray,Tb,Nb,Ab<:TracedRArray} + a::Tensor{Ta,Na,TracedRArray{Ta,Na}}, b::Tensor{Tb,Nb,TracedRArray{Tb,Nb}}; dims=(∩(inds(a), inds(b))), out=nothing +) where {Ta,Na,Tb,Nb} ia = collect(inds(a)) ib = collect(inds(b)) i = ∩(dims, ia, ib) @@ -159,7 +159,7 @@ function Tenet.contract( return _res end -function Tenet.contract(a::Tensor{T,N,A}; dims=nonunique(inds(a)), out=nothing) where {T,N,A<:TracedRArray} +function Tenet.contract(a::Tensor{T,N,TracedRArray{T,N}}; dims=nonunique(inds(a)), out=nothing) where {T,N} ia = inds(a) i = ∩(dims, ia)