tensorflow / recommenders-addons

Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.
Apache License 2.0
587 stars 132 forks source link

[fix] Supported hkv cpu operator base cuckoo cpu operator #387

Closed LinGeLin closed 5 months ago

LinGeLin commented 6 months ago

Because the underlying cuckoo gpu operator is changed to hkv operator, if a user originally uses cuckoo gpu training but does not have a gpu when serving, the inference service will fail。

Brief Description of the PR:

Fixes # (issue)

Type of change

Checklist:

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:

rhdong commented 5 months ago

Maybe we need to open some test cases for this PR.

LinGeLin commented 5 months ago

Maybe we need to open some test cases for this PR.

Some cpu test cases have been added, but if need ci to run these cases, it looks like need to add another ci node, is that necessary?