[Feature Request] - Liger Kernels on ARM64 for GH200?

[jlotthammer]

Hi, @arvindsun

I'm not sure if you've seen this repository from the folks at LinkedIn, but I was wondering on the off chance you did if you'd had any luck getting this to work on GH200? I've tried adding it to my container from nvidia using the builtin triton as you've done here, but I ran into issues with version conflicts of triton and its not clear to me why - looks like they require >2.3.0 and the NGC container I'm using ships with 3.0.0 so its.. befuzzling!

Their results at the surface look very impressive and I'm wondering if this is something you'd be interested in looking into as well and if you'd have better luck than I.

[arvindsun]

Yeah - we are looking at this too - plan to include it shortly.

[arvindsun]

Actually opened a PR linkedin/Liger-Kernel#79 - please comment on that