[Feature]: Can support for Strix halo(gfx1151) under windows？

[Jays-1111]

Suggestion Description

Is it supported on Strix Halo (gfx1151) running Windows

Windows

No response

GPU

Strix Halo gfx1151

ROCm Component

No response

[prabuselva]

+1 on this..
I understand triton is required for linux..
is it possible to compile CK backend on windows for strix halo ?

[rwfsmith]

I don't think it even supports strix halo in linux yet.

[Jays-1111]

I don't think it even supports strix halo in linux yet.

Ibelieve that Triton backend is ok on strixhalo

[Jays-1111]

+1 on this.. I understand triton is required for linux.. is it possible to compile CK backend on windows for strix halo ?

I don't think it is ok.

[rwfsmith]

This page has steps to install a 3.5.1 Triton wheel built for ROCm 7.1.1 on Linux, but I haven't seen an equivalent for Windows.

https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/install/installryz/native_linux/install-pytorch.html

There is something called AOTriton that can be enabled in Windows for the newer ROCm python builds, but I don't think it works the same as the separate Triton package?

Edit: there's now AMD support in the triton-windows package and I was able to follow the AMD specific steps in the main Flash Attention repo to build and install it. It kind of works, but some ops error out. Maybe somebody else will have better luck.