[Flaky Test] `tests/python/frontend/paddlepaddle/test_forward.py::test_forward_take_along_axis`

[WhatGhost]

These tests were found to be flaky (intermittently failing on main or failed in a PR with unrelated changes). See the docs for details.

Tests(s)

• tests/python/frontend/paddlepaddle/test_forward.py::test_forward_take_along_axis

Jenkins Links

• https://ci.tlcpack.ai/job/tvm-gpu/job/PR-14170/1/display/redirect

this test failed and the log show "AttributeError: module 'paddle' has no attribute 'take_along_axis'"
the take_alone_axis api was added since paddle2.3 ane i pass this test on my own env using paddle2.3.2.
how can i fix this error and pass the ci.
thanks

[WhatGhost]

[Frontend][Paddle]add take_alone_axis and topk converter for paddle frontend #14170

[jiangjiajun]

The ci image is still in paddlepaddle v2.1.3, we need to upgrade its version.

@masahi Could you help to update the docker image, I have send a pull request #14206

[masahi]

@jiangjiajun Sorry I was on vacation until today. Has anyone did the actual CI update since #14206? Otherwise I'll do it tomorrow.

[masahi]

I've uploaded a new GPU image https://hub.docker.com/layers/tlcpack/ci-gpu/20230308-070109-9d732d0fa/images/sha256-b14a65ebd3f96962241ceb949f4c1ee14fc4bb79bb4347fe5ed79c24044123c2?context=explore. Can you verify that it works for your test cases?

[masahi]

The new CPU and GPU images are being tested in https://github.com/apache/tvm/commits/ci-docker-staging

[masahi]

The CI images have been updated, please try again.