-
Notifications
You must be signed in to change notification settings - Fork 8
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
Update JAX and TFLite Models #166
Conversation
6037cb5
to
2084cfc
Compare
Comparative benchmarks is failing due to a compiler error in |
2084cfc
to
358c32c
Compare
@@ -13,6 +13,7 @@ on: | |||
schedule: | |||
# Scheduled to run at 09:00 UTC and 21:00 UTC. | |||
- cron: '0 09,21 * * *' | |||
pull_request: |
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.
Reminder of removal
@@ -248,7 +248,7 @@ jobs: | |||
run: | | |||
RESULTS_PATH="${RESULTS_DIR}/${JAX_IREE_RESULTS_JSON}" | |||
docker run --mount="type=bind,src="${PWD}",target=/work" --workdir="/work" \ | |||
"gcr.io/iree-oss/openxla-benchmark/base-python3.10@sha256:245a074284cfed5de60cf06a153e3bcd9a9c42702b6bb66a39bb47ef23b61669" \ | |||
"gcr.io/gcr.io/iree-oss/openxla-benchmark/base-python3.11@sha256:b9b98da7bcc5e431800ff798a6dcc394b1838a9ed3d695f5cd0dac3510fc8c8d" \ |
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.
gcr.io/gcr.io/iree-oss/
-> gcr.io/iree-oss/
?
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.
Ohh thank you for catching this!
358c32c
to
97ae086
Compare
There have been some recent changes to StableHLO that requires regeneration of JAX and TFLite models. TF model generation is currently broken due to version skews across HuggingFace, TF, and Keras.