Skip to content

Conversation

@snonk
Copy link

@snonk snonk commented Oct 19, 2025

No description provided.

location,
),
)
return res
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

construct a TracedRArray before returning (see the dot_general case below)

C.mlir_data,
alpha.mlir_data,
beta.mlir_data;
output=mlir_type(TracedRArray{eltype(C),length(ressize)}, ressize),
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

eltype(c) make this into unwrapped_eltype

β::Number=true,
)
# Promote to traced arrays
A = call_with_reactant(Reactant.promote_to, TracedRArray, A.data)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
A = call_with_reactant(Reactant.promote_to, TracedRArray, A.data)
A = call_with_reactant(Reactant.promote_to, TracedRArray, parent(A))

Comment on lines +277 to +279
@nospecialize(C::TracedRArray{T,2} where {T}),
@nospecialize(A::Symmetric),
@nospecialize(B::AbstractMatrix),
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also add the version where B is Symmetric

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants