Skip to content

Instantly share code, notes, and snippets.

@liuliu
Created October 7, 2023 23:49
Show Gist options
  • Save liuliu/54bbb000561820cba47c61058484e241 to your computer and use it in GitHub Desktop.
Save liuliu/54bbb000561820cba47c61058484e241 to your computer and use it in GitHub Desktop.
[graph dump]
"builtin.module"() ({
"func.func"() <{function_type = () -> (), sym_name = "main"}> ({
%0 = "mps.placeholder"() : () -> tensor<10x128xf32>
%1 = "mps.placeholder"() : () -> tensor<64x128xf32>
%2 = "mps.constant"() <{value = dense<0> : tensor<si32>}> : () -> tensor<si32>
%3 = "mps.constant"() <{value = dense<1> : tensor<si32>}> : () -> tensor<si32>
%4 = "mps.transpose"(%1, %2, %3) : (tensor<64x128xf32>, tensor<si32>, tensor<si32>) -> tensor<128x64xf32>
%5 = "mps.matmul"(%0, %4) <{transpose_lhs = false, transpose_rhs = false}> : (tensor<10x128xf32>, tensor<128x64xf32>) -> tensor<10x64xf32>
%6 = "mps.placeholder"() : () -> tensor<1x64xf32>
%7 = "mps.add"(%5, %6) : (tensor<10x64xf32>, tensor<1x64xf32>) -> tensor<10x64xf32>
}) : () -> ()
}) : () -> ()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment