-
Notifications
You must be signed in to change notification settings - Fork 3k
[INFER] Add cutlass fp8 gemm auto tune #9020
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
Conversation
Thanks for your contribution! |
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## develop #9020 +/- ##
========================================
Coverage 53.29% 53.29%
========================================
Files 652 652
Lines 105483 105483
========================================
Hits 56222 56222
Misses 49261 49261 ☔ View full report in Codecov by Sentry. |
@@ -0,0 +1,583 @@ | |||
# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
我建议在csrc/目录下建一个utils或者tools目录,把csrc/generate_code_dual_gemm_fused_kernels.py移进去,另外generate_code_dual_gemm_fused_kernels文件的命名也有点不太准确,改为auto_gen_fp8_fp8_dual_gemm_fused_kernels.py ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
另外顺便把csrc/test_tune_cublaslt_gemm.py也移到utils或者tools目录下,test_tune_cublaslt_gemm命名改为tune_cublaslt_int8_gemm.py吧
move to #9151 |
PR types
PR changes
Description