Skip to content

Adding backend_target for ModelType.init for working around TF-XLA#50

Merged
ganler merged 5 commits intomainfrom
fix-xla
Sep 19, 2022
Merged

Adding backend_target for ModelType.init for working around TF-XLA#50
ganler merged 5 commits intomainfrom
fix-xla

Conversation

@ganler
Copy link
Copy Markdown
Member

@ganler ganler commented Sep 19, 2022

TL;DR: Due to this, we need to use TensorFlow GPU-Eager for GPU-XLA and CPU-Eager for CPU-XLA. That information is passed via ModelType.init(..., backend_target=None|"cpu"|"gpu").

A few other refinements (basically ci) are also implemented. See commit history.

@ganler ganler merged commit 2b68ca5 into main Sep 19, 2022
@ganler ganler deleted the fix-xla branch September 19, 2022 05:30
@ganler ganler mentioned this pull request Sep 19, 2022
21 tasks
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.

1 participant