Skip to content

Commit fa4aeee

Browse files
authored
[CI] Update ci-gpu image (#15836)
* [CI] Update ci-gpu image The ci-gpu is updated to make changes in #15819 reflect in CI. * update image that contains oneflow==0.8.0 * Force old coreml api to fix failing test Between the previous ci_gpu image and the one in use now, the version of coreml has changed since it is unpinned (6.3.0 -> 7.0.0). This caused a test failure in the coreml frontend tests. The problem is that coreml now uses a new default format for models. This commit forces coreml to use the old model format so the test passes. More information can be found here: https://apple.github.io/coremltools/docs-guides/source/convert-learning-models.html
1 parent ec92ea9 commit fa4aeee

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

ci/jenkins/docker-images.ini

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
ci_arm: tlcpack/ci-arm:20230615-060132-62a5e7acf
2121
ci_cortexm: tlcpack/ci-cortexm:20230710-060128-a60cd0fec
2222
ci_cpu: tlcpack/ci-cpu:20230604-060130-0af9ff90e
23-
ci_gpu: tlcpack/ci-gpu:20230504-142417-4d37a0a0
23+
ci_gpu: tlcpack/ci-gpu:20231004-111300-7a1f7d0b
2424
ci_hexagon: tlcpack/ci-hexagon:20230504-142417-4d37a0a0
2525
ci_i386: tlcpack/ci-i386:20230504-142417-4d37a0a0
2626
ci_lint: tlcpack/ci-lint:20230504-142417-4d37a0a0

tests/python/frontend/coreml/test_forward.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -823,7 +823,9 @@ def test_can_build_keras_to_coreml_to_relay():
823823
kmodel_fn = path.join(tmpdir, "c1mdl.h5")
824824
model.save(kmodel_fn)
825825

826-
mdl = cm.convert(kmodel_fn)
826+
mdl = cm.convert(
827+
kmodel_fn, convert_to="neuralnetwork", minimum_deployment_target=cm.target.macOS11
828+
)
827829
model_file = path.join(tmpdir, "c1.mlmodel")
828830
mdl.save(model_file)
829831

0 commit comments

Comments
 (0)