Skip to content
This repository has been archived by the owner on Oct 25, 2024. It is now read-only.

Support starcoder MHA fusion #448

Merged
merged 4 commits into from
Oct 12, 2023
Merged

Support starcoder MHA fusion #448

merged 4 commits into from
Oct 12, 2023

Conversation

changwangss
Copy link
Contributor

@changwangss changwangss commented Oct 12, 2023

Type of Change

modify the model_gpt_bigcode.py to make mha fusion.
update benchmark info by token.
remaining to do, will update ipex installation after 2.1 release.

Description

detail description
JIRA ticket: xxx

Expected Behavior & Potential Risk

the expected behavior that triggered by this PR

How has this PR been tested?

how to reproduce the test (including hardware information)

Dependency Change?

any library dependency introduced or removed

Signed-off-by: changwangss <chang1.wang@intel.com>
Signed-off-by: changwangss <chang1.wang@intel.com>
Signed-off-by: changwangss <chang1.wang@intel.com>
@VincyZhang VincyZhang merged commit 841b29a into main Oct 12, 2023
2 of 7 checks passed
@VincyZhang VincyZhang deleted the wangchang/starcoder_mha branch October 12, 2023 10:15
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants