8000 Perfomance rule for `kron`. · Issue #526 · chalk-lab/Mooncake.jl · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Perfomance rule for kron. #526

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.

Sign up for GitHub

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

Jump to bottom
Open
yebai opened this issue Mar 16, 2025 · 0 comments
Open

Perfomance rule for kron. #526

yebai opened this issue Mar 16, 2025 · 0 comments
Assignees
@AstitvaAggarwal
Labels
enhancement: new rules required enhancement (performance) Would reduce the time it takes to run some bit of the code

Comments

@yebai
Copy link
Member
yebai commented Mar 16, 2025 •
edited
Loading

kron is the most underperforming benchmark function and requires a performance rule.

│                   kron_sum │ 377.0 μs │     39.9 │    12.9 │       220.0 │     7.2 │
│              kron_view_sum │ 364.0 μs │     43.5 │    9.97 │       208.0 │    78.8 │

The following Zygote rule can be adapted for Mooncake

https://github.com/FluxML/Zygote.jl/blob/c0bfc26de90db3778e1f3953f04d489e3413127b/src/lib/array.jl#L358-L374
@yebai yebai added enhancement (performance) Would reduce the time it takes to run some bit of the code enhancement: new rules required labels Mar 16, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@AstitvaAggarwal AstitvaAggarwal

Labels
enhancement: new rules required enhancement (performance) Would reduce the time it takes to run some bit of the code
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@yebai @AstitvaAggarwal
0