-
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
Add TF Dataset GPU #1354
Add TF Dataset GPU #1354
Conversation
Signed-off-by: Albert Wolant <awolant@nvidia.com>
CI MESSAGE: [935571]: BUILD STARTED |
CI MESSAGE: [935571]: BUILD FAILED |
Signed-off-by: Albert Wolant <awolant@nvidia.com>
CI MESSAGE: [937057]: BUILD STARTED |
CI MESSAGE: [937057]: BUILD FAILED |
Signed-off-by: Albert Wolant <awolant@nvidia.com>
CI MESSAGE: [937161]: BUILD STARTED |
CI MESSAGE: [937161]: BUILD FAILED |
CI MESSAGE: [937233]: BUILD STARTED |
CI MESSAGE: [937233]: BUILD FAILED |
CI MESSAGE: [937233]: BUILD PASSED |
!build |
CI MESSAGE: [937422]: BUILD STARTED |
CI MESSAGE: [937422]: BUILD PASSED |
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
CI MESSAGE: [940877]: BUILD STARTED |
CI MESSAGE: [940882]: BUILD STARTED |
device = "cpu", | ||
dtype = types.FLOAT16) | ||
device = device, | ||
dtype = types.INT16) |
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.
We don't support FLOAT16 in the Cast
on the GPU, so I've changed it to INT16 to have the same test both for the CPU and the GPU.
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.
I'll fix the Cast operator in a separate PR coming soon
CI MESSAGE: [940877]: BUILD PASSED |
Signed-off-by: Albert Wolant <awolant@nvidia.com>
CI MESSAGE: [940882]: BUILD PASSED |
CI MESSAGE: [941114]: BUILD STARTED |
CI MESSAGE: [941214]: BUILD STARTED |
CI MESSAGE: [941114]: BUILD FAILED |
CI MESSAGE: [941214]: BUILD PASSED |
CI MESSAGE: [941114]: BUILD PASSED |
CI MESSAGE: [943551]: BUILD STARTED |
CI MESSAGE: [943551]: BUILD PASSED |
Signed-off-by: Albert Wolant awolant@nvidia.com
Why we need this PR?
What happened in this PR?
Add GPU support for TF Dataset
Changes in
DALIDatasetOp
Added test for GPU and L1 test using other
device_id
JIRA TASK: [DALI-730]