[CI][VitisAI] Update CI Vitis AI PyXIR version to v0.3.1

[jtuyls]

This PR updates the CI Vitis AI PyXIR version to v0.3.1. This is needed for testing a subsequent PR that updates the Vitis AI docker and documentation to Vitis AI 1.4. The ReLU test is removed as it fails with the updated PyXIR version and will be added back in the other PR.
@comaniac @leandron @zhiics

[comaniac]

LGTM.
@areusch IIUC, we could now directly merge the image-related PRs and the image will be re-built nightly?

[tmoreau89]

LGTM, thank you @jtuyls !

[tmoreau89]

@comaniac @areusch is on PTO, I'll CC @leandron who can answer your question

[leandron]

Yes, the images are rebuilt daily, but not updated in CI automatically.

[leandron]

Does gen_requirements.py also needs to be updated with the correct version for vitis-ai (pyxir) dependency?

[jtuyls]

@leandron I can add the dependencies to the gen_requirements.py. I am not familiar with how it's set up but I guess it would look like this?

(
        "vitis-ai",
        (
            "Requirements for the Vitis AI integration",
            [
                "pyxir", 
                "h5py",
                "progressbar",
            ],
        ),
),

And is there anything else that needs to happen for this?

[leandron]

@leandron I can add the dependencies to the gen_requirements.py. I am not familiar with how it's set up but I guess it would look like this?

(
        "vitis-ai",
        (
            "Requirements for the Vitis AI integration",
            [
                "pyxir", 
                "h5py",
                "progressbar",
            ],
        ),
),

And is there anything else that needs to happen for this?

Hey @jtuyls, sorry it took me a bit of time to reply.

Yes, there is also the "constraints", in case you want to specify versions:

tvm/python/gen_requirements.py

Lines 204 to 206 in 8326edd

	CONSTRAINTS = [
	("astroid", None),
	("attrs", None),

Apart from that, your snippet above looks ok.

[jtuyls]

@leandron I added the dependencies to gen_requirements.py. I didn't add pyxir as it isn't pip installable.

[leandron]

LGTM, we just need to make sure that everywhere we use things that depend on pyxir, we lazy load and show and appropriate error message. Happy to merge this PR once it passes CI.