Skip to content

Instantly share code, notes, and snippets.

@fhanuman
fhanuman / split_and_sigmoid_dump_with_the_fix.mlir
Created April 29, 2026 21:01
PR #180: Full pipeline IR dump for split_sigmoid.onnx with --hip-promote-strided-operands enabled
// -----// IR Dump Before HipAddContextArgPass (hip-add-context-arg) ('builtin.module' operation) //----- //
module {
func.func @main_graph(%arg0: tensor<1x128x2048xf16> {onnx.name = "X"}) -> (tensor<1x128x1024xf16> {onnx.name = "outA"}, tensor<1x128x1024xf16> {onnx.name = "outB"}) attributes {onnx.graph.name = "split_sigmoid_test"} {
%0 = "onnx.NoValue"() {value} : () -> none
%1:2 = "onnx.Split"(%arg0) {axis = 2 : si64, node.outputs = ["A", "B"], onnx_node_name = ""} : (tensor<1x128x2048xf16>) -> (tensor<1x128x1024xf16>, tensor<1x128x1024xf16>)
%2 = "onnx.Sigmoid"(%1#0) {node.outputs = ["outA"], onnx_node_name = ""} : (tensor<1x128x1024xf16>) -> tensor<1x128x1024xf16>
%3 = "onnx.Sigmoid"(%1#1) {node.outputs = ["outB"], onnx_node_name = ""} : (tensor<1x128x1024xf16>) -> tensor<1x128x1024xf16>
"onnx.Return"(%2, %3) : (tensor<1x128x1024xf16>, tensor<1x128x1024xf16>) -> ()
}
}