-
Notifications
You must be signed in to change notification settings - Fork 4.4k
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
Improved options handling for TensorFlow sessions #41276
Conversation
+code-checks Logs: https://cmssdt.cern.ch/SDT/code-checks/cms-sw-PR-41276/35039
|
A new Pull Request was created by @valsdav (Davide Valsecchi) for master. It involves the following packages:
@cmsbuild, @mandrenguyen, @clacaputo can you please review it and eventually sign? Thanks. cms-bot commands are listed here |
Hi @clacaputo @mandrenguyen, just a kind ping for sending out tests for this technical PR... Thanks! |
please test |
+1 Summary: https://cmssdt.cern.ch/SDT/jenkins-artifacts/pull-request-integration/PR-1f822c/31929/summary.html Comparison SummarySummary:
|
+1 |
This pull request is fully signed and it will be integrated in one of the next master IBs (tests are also fine). This pull request will now be reviewed by the release team before it's merged. @perrotta, @dpiparo, @rappoccio (and backports should be raised in the release meeting by the corresponding L2) |
+1 |
@valsdav , I am testing TF 2.11 (cms-sw/cmsdist#8258) and looks like all the TF-CUdA tests are crashing see details https://cmssdt.cern.ch/SDT/jenkins-artifacts/pull-request-integration/PR-39ea0b/31966/unitTests/failed.html. I do see some warnings like [a] , any idea? [a]
|
Hi @smuzaffar I'm having a look. Thanks for testing TF2.11! |
PR description:
This PR is a follow up from PR #40551, #40551 (comment).
The PR improves the handling of the TF interface options by creating a new struct:
This change allows to have always a well defined default backend for TF sessions created with default options.
Moreover it simplifies the multiple overloaded functions for TF graph loading and session creation.
We will try to make the interface to ONNX more similar to the TF one in a separate PR, as in #39402
PR validation:
Technical PR: no changes on the reconstruction output are expected.
@riga @yongbinfeng