Skip to content
This repository was archived by the owner on Oct 16, 2023. It is now read-only.

Comments

Optimize Linear and GEGLU#152

Draft
mejai1206 wants to merge 1 commit intomainfrom
optimize_geglu
Draft

Optimize Linear and GEGLU#152
mejai1206 wants to merge 1 commit intomainfrom
optimize_geglu

Conversation

@mejai1206
Copy link
Contributor

@mejai1206 mejai1206 commented Sep 27, 2023

🙏 Describe the pull request

Linear와 GEGLU를 최적화하였습니다.

💬 Additional context

Batch가 있고, m이 작은 경우에 성능향상을 보입니다.

Linear (FP16, Batch=16, M=1, K=N)
image
image

GEGLU (FP16, Batch=16, M=1, K=N)
image
image

✅ Checklist

  • Code follows the project's coding conventions and style.
  • Tests have been added or updated to cover the changes.
  • Documentation has been updated, if necessary.

@mejai1206 mejai1206 force-pushed the optimize_geglu branch 2 times, most recently from c6908ec to 19230ac Compare September 27, 2023 02:28
@mejai1206 mejai1206 marked this pull request as draft September 27, 2023 02:38
@mejai1206 mejai1206 force-pushed the optimize_geglu branch 2 times, most recently from 8260887 to 67d4eb0 Compare October 4, 2023 04:53
@mejai1206 mejai1206 marked this pull request as ready for review October 4, 2023 05:32
Copy link
Contributor

@kakao-steve-ai kakao-steve-ai left a comment

Choose a reason for hiding this comment

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

LGTM

@mejai1206 mejai1206 marked this pull request as draft October 5, 2023 07:29
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