AMDMIGraphX
AMDMIGraphX copied to clipboard
Add support for 'SplitToSequence' ONNX operator.
DOR (Definition of Ready)
No extra preparations are needed. 'SplitToSequence' is a documented ONNX operator.
Description
FLUX model requires MIGraphX to support 'SplitToSequence' ONNX operator since 'diffusers' version 0.35.0. Probably, it is needed for mapping aten::rms_norm operation into ONNX graph, but I couldn't find confirmation for it in 'diffusers' github.
One needs to support 'SplitToSecond' operator in MIGraphX.
Operator description: https://github.com/onnx/onnx/blob/main/docs/Operators.md#splittosequence
DOD (Definition of Done)
FLUX compilation script works without errors with ONNX opset version 23+ AND with 'diffusers' module of version 0.35.0 or higher. Compiled model produces outputs that are close enough to non-compiled model.