Added triton interop support

[jamesthejellyfish]

added the ability to call triton functions from within a PyDSL function. This works by parsing the triton ir (ttir), and then using an open-source tool called triton-adapter-opt from triton-ascend to convert the ttir to linalg mlir, which then gets combined into the PyDSL mlir file. Note that when calling a triton function, in addition to the triton args, you must include three extra UInt32 parameters which define the grid index (x,y,z)

[Balint-R]

I left some comments. I have not really looked at test_triton.py or examples/triton_interop.py yet. I will review this again after most of my comments are resolved.

My biggest question right now is probably, what is the code that parses how many arguments the Triton function has based on the MLIR and why do we need it? Also in general, how does the process of actually calling the Triton function work?

[Balint-R]

We should also figure out how to set up Triton in CI.

[jamesthejellyfish]

We should also figure out how to set up Triton in CI.

Yes I have been talking with @xushengj about this. It will probably be something I do at the same time when I am working on the feature to separate the formatting and the test on CI.

[Balint-R]

Comments I marked as resolved but we should revisit later:

• https://github.com/Huawei-CPLLab/PyDSL/pull/36/files#r2258102148 triton-adapter-opt converts Index to UInt32. Complain to triton-adapter-opt.
• https://github.com/Huawei-CPLLab/PyDSL/pull/36/files#r2257480284 Figure out return type of Triton function. Future PR: reverse parse ttir string to MLIR type.
• https://github.com/Huawei-CPLLab/PyDSL/pull/36/files#r2257337640 Does Triton allow keyword arguments? Future PR: support kw args for Triton functions.
• Added triton interop support #36 (comment) PR: Move grid indexing to square bracket notation. Then slightly improve the check of whether num args is correct.

[Balint-R]

Maybe address my existing comments and #63, and then we can merge. I am tired of reading over this so many times. There are probably not major issues anymore.

[Balint-R]

@jamesthejellyfish you should fix the one thing I commented today, rebase onto main (you will have to resolve a trivial merge conflict), then you can merge!