-
Notifications
You must be signed in to change notification settings - Fork 615
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
Input device check #5631
Input device check #5631
Conversation
!build |
CI MESSAGE: [18307567]: BUILD STARTED |
CI MESSAGE: [18305355]: BUILD FAILED |
!build |
CI MESSAGE: [18309171]: BUILD STARTED |
CI MESSAGE: [18309171]: BUILD FAILED |
CI MESSAGE: [18324189]: BUILD STARTED |
CI MESSAGE: [18324189]: BUILD FAILED |
* Add experimental_exec_dynamic flag to Pipeline to enable new executor * Add DataNode.cpu() that triggers a GPU->CPU copy * Remove checks that prevented GPU->CPU transitions from Python and Pipeline class * Remove checks that prevented CPU operators from taking GPU inputs * Use old executor's graph lowering to run the checks * Add cpu->gpu tests * TODO: Improve input backend checks (#5631) * TODO: Add tensorflow support --------- Signed-off-by: Michal Zientkiewicz <[email protected]>
Signed-off-by: Michał Zientkiewicz <[email protected]>
Signed-off-by: Michał Zientkiewicz <[email protected]>
Signed-off-by: Michal Zientkiewicz <[email protected]>
1b7aebc
to
6ad76f7
Compare
Signed-off-by: Michał Zientkiewicz <[email protected]>
bool has_shape_, has_dtype_; | ||
DALIDataType dtype_; | ||
|
||
bool has_const_value_ = false; | ||
int const_value_ = 0; | ||
|
||
const int kShapeLikeInputIdx = is_shape_like_ ? 0 : -1; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
These are not compile-time kOnstants - just regular fields.
CI MESSAGE: [18338618]: BUILD STARTED |
CI MESSAGE: [18338618]: BUILD FAILED |
Signed-off-by: Michał Zientkiewicz <[email protected]>
CI MESSAGE: [18356254]: BUILD STARTED |
CI MESSAGE: [18356254]: BUILD FAILED |
CI MESSAGE: [18433207]: BUILD STARTED |
Signed-off-by: Michal Zientkiewicz <[email protected]>
fbb1b08
to
2e89b84
Compare
CI MESSAGE: [18433770]: BUILD STARTED |
CI MESSAGE: [18433770]: BUILD PASSED |
* Check input device in OpSpec::AddInput. * Add `Any` and `MatchBackendOrCPU` input devices * Fix InputDevice in operators. Add Any device capability to "shape_like" inputs. * Add python-side backend validation in logical expressions. --------- Signed-off-by: Michal Zientkiewicz <[email protected]>
Category:
New feature (non-breaking change which adds functionality)
Description:
This PR extends input device specification in OpSchema and adds input device validateion to OpSpec.
Additional information:
Affected modules and functionalities:
Key points relevant for the review:
Tests:
Some tests were modified.
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A