Skip to content
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

Merged
merged 2 commits into from
Dec 8, 2023
Merged

Conversation

sampathweb
Copy link
Collaborator

Adds Kokoro GPU Tests similar to Keras-NLP and Keras

Copy link
Member

@mattdangerw mattdangerw left a 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 \
Copy link
Member

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)

Copy link
Collaborator Author

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

@sampathweb sampathweb merged commit da60e9d into keras-team:master Dec 8, 2023
6 of 9 checks passed
@sampathweb sampathweb deleted the kokoro-gpu-tests branch December 8, 2023 22:40
divyashreepathihalli pushed a commit that referenced this pull request Dec 11, 2023
* 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>
yuvraj-wale pushed a commit to yuvraj-wale/keras-cv that referenced this pull request Feb 8, 2024
* Adds Kokoro tests

* Add Kokoro Tests
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.

None yet

2 participants