Update Flashinfer from v0.4.1 to v0.5.2

[hmellor]

• Bump Flashinfer to v0.5.2
• Remove workaround from docker build
• Allows us to stop needing --pre when installing from source (fixes [Installation]: FlashInfer Dependency issue due to pre-release apache-tvm-ffi #27476)

N.B. xformers is also causing --pre to be required at the moment

[gemini-code-assist]

Code Review

This pull request updates Flashinfer from version v0.4.1 to v0.5.0. The changes include updating the package versions in requirements/cuda.txt and the Dockerfiles, as well as removing a related workaround that is no longer necessary. The modifications are consistent and correctly implement the version bump. The changes look good.

[youkaichao]

cc @mgoin @pavanimajety

[hmellor]

Both Blackwell tests passed in last night's nightly. So these appear to be new and legitimate failures.

[lgeiger]

Shall we try whether v0.5.1 fixed it?

[hmellor]

The changelog doesn't look like it fixes any bugs. We have a fix for the unquantised test (relaxing the tolerances), but I'm still waiting for a solution for the quantised moe test.

[hmellor]

MoE investigation here: flashinfer-ai/flashinfer#2032

[hmellor]

flashinfer-ai/flashinfer#2049 contains the MoE fix and was included in 0.5.2 released 1h ago

[yewentao256]

LGTM, thanks for the work!
Could we trigger the full tests (including all optional) for this change?

[hmellor]

Changing the requirements/dockerfile already does trigger full CI.

There are some optional ones that are only run nightly, but many of those are failing already and so would block this PR from merging.