-
Notifications
You must be signed in to change notification settings - Fork 2.9k
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
MlasTranspose multi-threads support. #22912
base: main
Are you sure you want to change the base?
MlasTranspose multi-threads support. #22912
Conversation
@msy-kato please read the following Contributor License Agreement(CLA). If you agree with the CLA, please reply with the following information.
Contributor License AgreementContribution License AgreementThis Contribution License Agreement (“Agreement”) is agreed to by the party signing below (“You”),
|
Description
MlasTranspose was running single-thread and was not performing well enough on a multi-threaded CPU. Therefore, we modified it to run with multi-thread to improve performance.
The
MlasTranspose
was previously running in a single-threaded, which resulted in suboptimal performance on multi-threaded CPUs. To address this, we have modified it to utilize multi-threading.Motivation and Context
We encountered this issue while running the multilingual-e5-large, which was converted to ONNX format and executed on a multi-core CPU (Xeon 6338T). Below are the performance metrics before and after the modification:
Condition
This change resulted in a performance improvement of approximately 14%(
MlasTranspose
is x2~3 faster).