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

[Feat]combine cpu and gpu wheel #456

Merged
merged 1 commit into from
Sep 2, 2024
Merged

Conversation

jq
Copy link
Collaborator

@jq jq commented Aug 28, 2024

Description

Brief Description of the PR:
combine cpu and gpu wheel so that you only need pip install tensorflow-recommenders-addons
Fixes # (issue)

Type of change

  • Bug fix
  • New Tutorial
  • Updated or additional documentation
  • Additional Testing
  • New Feature

Checklist:

  • I've properly formatted my code according to the guidelines
    • By running yapf
    • By running clang-format
  • This PR addresses an already submitted issue for TensorFlow Recommenders-Addons
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works

How Has This Been Tested?

If you're adding a bugfix or new feature please describe the tests that you ran to verify your changes:
*

@jq jq force-pushed the hkv_hashtable_ops branch 2 times, most recently from 36ea21a to 55a594e Compare August 29, 2024 00:23
@rhdong rhdong changed the title [New Feature]combine cpu and gpu wheel [Feat]combine cpu and gpu wheel Aug 30, 2024
rhdong
rhdong previously approved these changes Aug 30, 2024
Copy link
Member

@rhdong rhdong left a comment

Choose a reason for hiding this comment

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

LGTM

README.md Outdated
@@ -76,7 +76,7 @@ run the following:
pip install tensorflow-recommenders-addons
```

By default, CPU version will be installed. To install GPU version, run the following:
Before the release of combined CPU and GPU version, to install GPU version, run the following:
Copy link
Member

Choose a reason for hiding this comment

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

Better to explain more clearly, for example, the specific version(0.7.x) after which we make this change.

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

updated

@rhdong
Copy link
Member

rhdong commented Aug 30, 2024

Hey @MoFHeka pls review it when you feel convenient, basically, it's LGTM.

@jq jq force-pushed the hkv_hashtable_ops branch from c3e7f0c to 39bca95 Compare August 31, 2024 00:50
Copy link
Collaborator

@MoFHeka MoFHeka left a comment

Choose a reason for hiding this comment

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

LGTM

@rhdong rhdong merged commit 9384a0a into tensorflow:master Sep 2, 2024
41 checks passed
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.

3 participants