This is an automated email from the ASF dual-hosted git repository.
ashutoshp pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new fa4aeee64e [CI] Update ci-gpu image (#15836)
fa4aeee64e is described below
commit fa4aeee64efbd55db1f502a220276f9851c52f15
Author: Luke Hutton <[email protected]>
AuthorDate: Fri Oct 6 09:35:32 2023 +0100
[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
---
ci/jenkins/docker-images.ini | 2 +-
tests/python/frontend/coreml/test_forward.py | 4 +++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/ci/jenkins/docker-images.ini b/ci/jenkins/docker-images.ini
index 78f24dcbd3..3bbf5b90d6 100644
--- a/ci/jenkins/docker-images.ini
+++ b/ci/jenkins/docker-images.ini
@@ -20,7 +20,7 @@
ci_arm: tlcpack/ci-arm:20230615-060132-62a5e7acf
ci_cortexm: tlcpack/ci-cortexm:20230710-060128-a60cd0fec
ci_cpu: tlcpack/ci-cpu:20230604-060130-0af9ff90e
-ci_gpu: tlcpack/ci-gpu:20230504-142417-4d37a0a0
+ci_gpu: tlcpack/ci-gpu:20231004-111300-7a1f7d0b
ci_hexagon: tlcpack/ci-hexagon:20230504-142417-4d37a0a0
ci_i386: tlcpack/ci-i386:20230504-142417-4d37a0a0
ci_lint: tlcpack/ci-lint:20230504-142417-4d37a0a0
diff --git a/tests/python/frontend/coreml/test_forward.py
b/tests/python/frontend/coreml/test_forward.py
index dd1aaf3085..e381f44734 100644
--- a/tests/python/frontend/coreml/test_forward.py
+++ b/tests/python/frontend/coreml/test_forward.py
@@ -823,7 +823,9 @@ def test_can_build_keras_to_coreml_to_relay():
kmodel_fn = path.join(tmpdir, "c1mdl.h5")
model.save(kmodel_fn)
- mdl = cm.convert(kmodel_fn)
+ mdl = cm.convert(
+ kmodel_fn, convert_to="neuralnetwork",
minimum_deployment_target=cm.target.macOS11
+ )
model_file = path.join(tmpdir, "c1.mlmodel")
mdl.save(model_file)
