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

Implement Fold and Unfold #3167

Open
wants to merge 74 commits into
base: develop
Choose a base branch
from
Open

Implement Fold and Unfold #3167

wants to merge 74 commits into from

Conversation

DuongQLee
Copy link
Collaborator

@DuongQLee DuongQLee commented Jul 29, 2024

  • Added Fold and Unfold op.
  • Full benchmark result compared to ROCm Here
  • Average performance:

Op Dtype Direction Time
Unfold fp32 fwd 16.43641979
fp32 bwd 1.458798342
fp16 fwd 15.83955361
fp16 bwd 1.459763543
bfp16 fwd 15.90593279
bfp16 bwd 1.455323877
Fold fp32 fwd 1.463731927
fp32 bwd 28.09828887
fp16 fwd 1.479315364
fp16 bwd 28.08557933
bfp16 fwd 1.47158127
bfp16 bwd 26.84660993

@hieule88 hieule88 requested a review from CAHEK7 October 18, 2024 04:10
@hieule88 hieule88 assigned hieule88 and unassigned DuongQLee Oct 18, 2024
@hieule88
Copy link
Collaborator

@iq136boy Would you send us the build log of this PR?

Copy link
Contributor

@iq136boy iq136boy left a comment

Choose a reason for hiding this comment

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

Please fix the following errors:
error_log_3167.txt

Copy link
Contributor

@iq136boy iq136boy left a comment

Choose a reason for hiding this comment

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

Error log:
error_log (2).txt

Copy link
Contributor

@iq136boy iq136boy left a comment

Choose a reason for hiding this comment

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

Error log:
3167_log.txt

@hieule88
Copy link
Collaborator

Error log: 3167_log.txt

I fixed this error. Please send me the newest error log of this PR.

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.

5 participants