Does rocm/flashinfer have something similar to flashinfer.green_ctx?

[monopodium]

I found that there is no green_ctx as in nvidia flashinfer. Are there any other modules from rocm-flashinfer that implement "spatial sharing GPU SM" features?

[diptorupd]

@monopodium thank you for reaching out, our ROCm fork is still in very early stage of development and we are not at par with upstream CUDA flashinfer.

Based on my quick review of upstream the green_ctx feature landed into upstream flashinfer in v0.2.7 and we are at this moment tracking an older release. We will update the upstream release we track as soon as some of the core features we are working towards (as mentioned in the README) are done.

Moreover, I need to investigate if the ROCm stack has an equivalent to CUDA green context. Let me do some research and get back.