Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEA] Add outer() for outer product #509

Closed
cliffburdick opened this issue Oct 31, 2023 · 2 comments
Closed

[FEA] Add outer() for outer product #509

cliffburdick opened this issue Oct 31, 2023 · 2 comments

Comments

@cliffburdick
Copy link
Collaborator

Use cuBLAS *ger functions

@luitjens
Copy link
Collaborator

luitjens commented Jan 5, 2024

If we have something like outer_impl(A, X, Y) where A is batch_dims x N x M and X is batch_dims x N and y is batch_dims x M then we should be able to do this pretty easily by having outer internally clone X to X is batch_dims x N x 1 and clone Y to batch_dims x 1 x M then call matmul_impl.

Note we should probably also support the axis parameter to match matmul/matvec.

@luitjens
Copy link
Collaborator

@cliffburdick fixed?

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

No branches or pull requests

2 participants