-
Notifications
You must be signed in to change notification settings - Fork 322
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
Adds Kokoro GPU Tests #2224
Adds Kokoro GPU Tests #2224
Conversation
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.
Looks good!
# Run Extra Large Tests for Continuous builds | ||
if [ "${RUN_XLARGE:-0}" == "1" ] | ||
then | ||
pytest --check_gpu --run_large --run_extra_large --durations 0 \ |
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.
General question, why are we doing so much enumeration of testing here. Seems hard to maintain. Why not just run it all?
(probably good to land this equivalent to before, just have been curious about this)
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 kept it same as cloudbuild
config. We should open it all. But that's how the GPU cloudbuild is setup so copied the command as-is
* Adds Kokoro GPU Tests (#2224) * Adds Kokoro tests * Add Kokoro Tests * Update random_cutout.py to be a subclass of VectorizedBaseImageAugmentationLayer (#2123) * Update random_cutout.py Make RandomCutout a subclass of VectorizedBaseImageAugmentationLayer * Fix RandomCutout for ragged inputs * Fix typo and style * Vectorized implementation of RandomColorDegeneration and Equalization preprocessing layers (#2214) * Vectorize RandomColorDegeneration * Vectorize Equalization * Vectorize Equalization * Fix Equalization for ragged input --------- Co-authored-by: Ramesh Sampath <1437573+sampathweb@users.noreply.github.com> Co-authored-by: SUPERGIU <supergiu@outlook.com>
* Adds Kokoro tests * Add Kokoro Tests
Adds Kokoro GPU Tests similar to Keras-NLP and Keras