Skip to content

improvements to tests and right matmul error#84

Open
Transurgeon wants to merge 1 commit intomainfrom
param-left-matmul
Open

improvements to tests and right matmul error#84
Transurgeon wants to merge 1 commit intomainfrom
param-left-matmul

Conversation

@Transurgeon
Copy link
Copy Markdown
Collaborator

This PR adds three things:

  • new tests for param forward in left_matmul
  • new tests for forward of right_matmul
  • some error validation for right_matmul for its arguments.. rather than delegating to left_matmul

I encountered these things while testing the diffengine backend.
The fix is actually upstream in the converters, but I thought I would add some tests to the diffengine to make it a bit more robust.
The right matmul bug was encountered when doing (1,n) @ (n) upstream, and the right hand argument's shape would be incorrectly converted to (1,n). I now make it so there's a special case to detect when a 1d array appears in either a LHS or RHS of a matmul.

@dance858
Copy link
Copy Markdown
Collaborator

When chatting offline we decided to revisit this PR

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