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

Use cuBLASlt in attention #325

Merged
merged 18 commits into from
May 18, 2024
Merged

Use cuBLASlt in attention #325

merged 18 commits into from
May 18, 2024

Conversation

EricLBuehler
Copy link
Owner

No description provided.

Copy link

Code Metrics Report
  ===============================================================================
 Language            Files        Lines         Code     Comments       Blanks
===============================================================================
 Dockerfile              1           34           25            0            9
 Happy                   1          442          369            0           73
 JSON                    5            9            9            0            0
 Python                 21          741          622           21           98
 TOML                   15          390          352            1           37
-------------------------------------------------------------------------------
 Jupyter Notebooks       1            0            0            0            0
 |- Markdown             1           60           30           22            8
 |- Python               1           96           87            1            8
 (Total)                            156          117           23           16
-------------------------------------------------------------------------------
 Markdown               15         1026            0          758          268
 |- BASH                 6          205          192            0           13
 |- Python               6          121          110            0           11
 |- Rust                 3          185          172            9            4
 (Total)                           1537          474          767          296
-------------------------------------------------------------------------------
 Rust                   80        26617        24483          340         1794
 |- Markdown            38          382            0          377            5
 (Total)                          26999        24483          717         1799
===============================================================================
 Total                 140        29259        25860         1120         2279
===============================================================================
  

Copy link
Owner Author

@EricLBuehler EricLBuehler left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good.

@EricLBuehler
Copy link
Owner Author

This PR gives +12% performance improvement over master (7da5b71).

@EricLBuehler EricLBuehler merged commit 5a2a577 into master May 18, 2024
11 checks passed
@EricLBuehler EricLBuehler deleted the cublaslt branch May 18, 2024 14:22
@EricLBuehler EricLBuehler restored the cublaslt branch May 29, 2024 13:51
@EricLBuehler EricLBuehler deleted the cublaslt branch May 29, 2024 13:51
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.

None yet

1 participant