Skip to content

Conversation

@masahi
Copy link
Member

@masahi masahi commented Mar 31, 2022

To support the new PT fallback feature, we need to update PT to 1.11 (for dlpack bump). See more discussion in #10758

Validated on https://ci.tlcpack.ai/blue/organizations/jenkins/tvm/detail/ci-docker-staging/244/pipeline

@t-vi @driazati @areusch @leandron @Mousius

@masahi masahi force-pushed the ci-gpu-torch-1.11 branch from ecd4ba2 to 02c9df9 Compare March 31, 2022 19:53
from matplotlib import pyplot as plt
from keras.applications.resnet50 import preprocess_input
from tensorflow.keras.applications.resnet50 import preprocess_input

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't get why I had to make this change. After we upgraded to keras 2.6 in #10084, keras.applications no longer exists. The doc build runs on the gpu image, which should have been updated to keras 2.6 in #10654. So this error should have been caught during the update of #10654.

cc @driazati @leandron

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

also not sure, the mapping of image version -> range of commits is very difficult to track down (but should be fixed by #10646)

from matplotlib import pyplot as plt
from keras.applications.resnet50 import preprocess_input
from tensorflow.keras.applications.resnet50 import preprocess_input

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

also not sure, the mapping of image version -> range of commits is very difficult to track down (but should be fixed by #10646)

@mbrookhart mbrookhart merged commit 621f777 into main Apr 1, 2022
@masahi masahi deleted the ci-gpu-torch-1.11 branch April 4, 2022 10:40
pfk-beta pushed a commit to pfk-beta/tvm that referenced this pull request Apr 11, 2022
* Update Jenkinsfile to point to the new GPU image

* fix onnx test

* fixed keras tutorial
mehrdadh pushed a commit to mehrdadh/tvm that referenced this pull request Apr 11, 2022
* Update Jenkinsfile to point to the new GPU image

* fix onnx test

* fixed keras tutorial
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.

3 participants