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

remove autocast for cpu #491

Merged
merged 4 commits into from
Jun 13, 2023
Merged

remove autocast for cpu #491

merged 4 commits into from
Jun 13, 2023

Conversation

wanliAlex
Copy link
Collaborator

  • What kind of change does this PR introduce? (Bug fix, feature, docs update, ...)
    remove autocast for cpu

  • What is the current behavior? (You can also link to an open issue here)
    we have autocast for CPU, which make arm64 vectorise speed slow

  • What is the new behavior (if this is a feature change)?
    remove autocast for CPU now

  • Does this PR introduce a breaking change? (What changes might users need to make in their application due to this PR?)
    no

  • Have unit tests been run against this PR? (Has there also been any additional testing?)
    no

  • Related Python client changes (link commit/PR here)
    no

  • Related documentation changes (link commit/PR here)
    no

  • Other information:
    no

  • Please check if the PR fulfills these requirements

  • The commit message follows our guidelines
  • Tests for the changes have been added (for bug fixes/features)
  • Docs have been added / updated (for bug fixes / features)

@wanliAlex wanliAlex temporarily deployed to marqo-test-suite May 24, 2023 10:45 — with GitHub Actions Inactive
@wanliAlex wanliAlex temporarily deployed to marqo-test-suite May 24, 2023 10:45 — with GitHub Actions Inactive
@wanliAlex wanliAlex temporarily deployed to marqo-test-suite May 24, 2023 10:45 — with GitHub Actions Inactive
@wanliAlex wanliAlex temporarily deployed to marqo-test-suite May 24, 2023 10:45 — with GitHub Actions Inactive
Copy link
Collaborator

@pandu-k pandu-k left a comment

Choose a reason for hiding this comment

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

have you tested that this change improves arm64 vectorising speed?

@wanliAlex
Copy link
Collaborator Author

have you tested that this change improves arm64 vectorising speed?

yes, tested both on my mac and ec2 instance.

@pandu-k pandu-k requested a review from jn2clark June 9, 2023 00:31
@wanliAlex wanliAlex temporarily deployed to marqo-test-suite June 12, 2023 14:02 — with GitHub Actions Inactive
@pandu-k pandu-k merged commit dc0fdae into mainline Jun 13, 2023
@pandu-k pandu-k deleted the li/autocast-fix branch June 13, 2023 00:07
@wanliAlex
Copy link
Collaborator Author

image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants