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

DArray: Implement efficient in-place matrix-matrix multiply #492

Merged
merged 20 commits into from
Apr 5, 2024

Conversation

Rabab53
Copy link
Collaborator

@Rabab53 Rabab53 commented Mar 26, 2024

This PR implements:

  • Optimized mul! and * using datadeps
  • Optimized transpose/adjoint using datadeps
  • copyto! using datadeps
  • Optimized getindex with optional task-local caching
  • setindex! (to be optimized later)
  • A convenient test runner script (used like julia test/runtests.jl --test array/linalg/matmul --test array/copies)

@jpsamaroo jpsamaroo force-pushed the matmul branch 2 times, most recently from 75233f4 to 5fc5cf0 Compare March 26, 2024 20:50
@jpsamaroo jpsamaroo changed the title Matmul DArray: Implement efficient in-place matrix-matrix multiply Mar 26, 2024
@jpsamaroo jpsamaroo force-pushed the matmul branch 4 times, most recently from 5d31832 to 2e0decb Compare April 3, 2024 18:29
@jpsamaroo jpsamaroo force-pushed the matmul branch 5 times, most recently from e3b2632 to 0f14a43 Compare April 5, 2024 14:22
@jpsamaroo jpsamaroo force-pushed the matmul branch 2 times, most recently from 587ae6e to 3c168b2 Compare April 5, 2024 19:32
Adds optional task-local cache for getindex
Adds setindex! operation
Fixes lookup_parts indexing fallback for linear indexing
Adds allowscalar logic to prevent slow paths
Adds first/last helpers
@jpsamaroo jpsamaroo merged commit 541c3bd into JuliaParallel:master Apr 5, 2024
9 of 12 checks passed
@jpsamaroo
Copy link
Member

Thanks @Rabab53!

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

Successfully merging this pull request may close these issues.

2 participants