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鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

馃殌 perf: Convolution2D #118

Merged
merged 10 commits into from
Feb 28, 2024
Merged

馃殌 perf: Convolution2D #118

merged 10 commits into from
Feb 28, 2024

Conversation

jean-francoisreboud
Copy link
Collaborator

@jean-francoisreboud jean-francoisreboud commented Feb 27, 2024

Changes:

  • some kernel optimization: locality

Benchmark:

20 iterations of 64 images of size (224, 224)

After some kernel optimizations:

M3Max 6900XT
Train VGG16 119s (74% faster) 145s (66% faster)
Eval VGG16 38s (48% faster) 30s (39% faster)

Before kernel optimizations:

M3Max 6900XT
Train VGG16 455s 422s
Eval VGG16 73s 49s

@jean-francoisreboud jean-francoisreboud self-assigned this Feb 27, 2024
@jean-francoisreboud jean-francoisreboud merged commit 192f994 into release_5 Feb 28, 2024
3 checks passed
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.

None yet

1 participant