pashu123
/ AtenComplexRealModule.mlir
Created
April 25, 2023 16:12
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #loc = loc(unknown) | |
| #map = affine_map<(d0) -> (d0)> | |
| module attributes {torch.debug_module_name = "AtenComplexRealModule"} { | |
| ml_program.global private mutable @global_seed(dense<0> : tensor<i64>) : tensor<i64> loc(#loc) | |
| func.func @forward(%arg0: tensor<?xcomplex<f32>> loc(unknown)) -> tensor<?xf32> { | |
| %c0 = arith.constant 0 : index loc(#loc) | |
| %dim = tensor.dim %arg0, %c0 : tensor<?xcomplex<f32>> loc(#loc) | |
| %0 = tensor.empty(%dim) : tensor<?xf32> loc(#loc) | |
| %1 = linalg.generic {indexing_maps = [#map, #map], iterator_types = ["parallel"]} ins(%arg0 : tensor<?xcomplex<f32>>) outs(%0 : tensor<?xf32>) { | |
| ^bb0(%in: complex<f32> loc(unknown), %out: f32 loc(unknown)): |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| from diffusers import StableDiffusionPipeline | |
| import torch_mlir | |
| from shark.shark_importer import import_with_fx | |
| import os | |
| import torch.fx as fx | |
| import sys | |
| model_input = { |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| graph(): | |
| %arg0_1 : [#users=1] = placeholder[target=arg0_1] | |
| %arg1_1 : [#users=1] = placeholder[target=arg1_1] | |
| %arg2_1 : [#users=32] = placeholder[target=arg2_1] | |
| %expand : [#users=1] = call_function[target=torch.ops.aten.expand](args = (%arg1_1, [2]), kwargs = {}) | |
| %arange : [#users=1] = call_function[target=torch.ops.aten.arange](args = (0, 160), kwargs = {dtype: torch.float32, device: cuda:0, pin_memory: False}) | |
| %mul : [#users=1] = call_function[target=torch.ops.aten.mul](args = (%arange, -9.210340371976184), kwargs = {}) | |
| %div : [#users=1] = call_function[target=torch.ops.aten.div](args = (%mul, 160), kwargs = {}) | |
| %exp : [#users=1] = call_function[target=torch.ops.aten.exp](args = (%div,), kwargs = {}) | |
| %slice_1 : [#users=1] = call_function[target=torch.ops.aten.slice](args = (%expand, 0, 0, 9223372036854775807), kwargs = {}) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| from diffusers import StableDiffusionPipeline | |
| import torch_mlir | |
| from shark.shark_importer import import_with_fx | |
| import os | |
| import torch.fx as fx | |
| import sys | |
| model_input = { |
This file has been truncated, but you can view the full file.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #composite_of_1731821440b = #util.composite<1731821440xi8, [ | |
| dense_resource<__elided__> : tensor<320x1280xf16>, | |
| dense_resource<__elided__> : tensor<1280x1280xf16>, | |
| dense_resource<__elided__> : tensor<1280x320xf16>, | |
| dense_resource<__elided__> : tensor<320x320xf16>, | |
| dense_resource<__elided__> : tensor<320x320xf16>, | |
| dense_resource<__elided__> : tensor<320x320xf16>, | |
| dense_resource<__elided__> : tensor<320x320xf16>, | |
| dense_resource<__elided__> : tensor<320x320xf16>, | |
| dense_resource<__elided__> : tensor<320x320xf16>, |
pashu123
/ gist:e76af4aac1cdd30afa5339add2506e51
Created
March 31, 2023 15:43
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| ========== | |
| VULKANINFO | |
| ========== | |
| Vulkan Instance Version: 1.3.231 | |
| Instance Extensions: count = 19 | |
| =============================== | |
| VK_EXT_acquire_xlib_display : extension revision 1 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| from diffusers import StableDiffusionPipeline | |
| import torch_mlir | |
| from shark.shark_importer import import_with_fx | |
| import os | |
| model_input = { | |
| "clip": (torch.randint(1, 2, (1, 77)),), | |
| "vae": (torch.randn(1, 4, 128, 128),), |
pashu123
/ int8_elided.mlir
Created
March 20, 2023 15:58
This file has been truncated, but you can view the full file.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #map = affine_map<(d0, d1, d2, d3) -> (0, d1, d2, d3)> | |
| #map1 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)> | |
| #map2 = affine_map<(d0) -> (0)> | |
| #map3 = affine_map<(d0) -> (d0)> | |
| #map4 = affine_map<(d0) -> ()> | |
| #map5 = affine_map<(d0, d1) -> (d0, 0)> | |
| #map6 = affine_map<(d0, d1) -> (0, d1)> | |
| #map7 = affine_map<(d0, d1) -> (d0, d1)> | |
| #map8 = affine_map<(d0, d1) -> (d1, d0)> | |
| #map9 = affine_map<(d0, d1) -> (d1)> |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from iree import runtime as ireert | |
| from iree.compiler import compile_str | |
| import numpy as np | |
| LINALG_IR = ''' | |
| #map = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)> | |
| module { | |
| func.func @forward(%arg0: tensor<1x6x32x64x2xf32>) -> tensor<1x6x32x64xcomplex<f32>> { | |
| %c1 = arith.constant 1 : index |
pashu123
/ llama_linalg.mlir
Created
March 3, 2023 15:38
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| func.func @forward(%arg0: tensor<512xf32>, %arg1: tensor<512xf32>, %arg2: tensor<512xf32>, %arg3: tensor<512xf32>, %arg4: tensor<512xf32>, %arg5: tensor<512xf32>, %arg6: tensor<512xf32>, %arg7: tensor<512xf32>, %arg8: tensor<512xf32>, %arg9: tensor<512xf32>, %arg10: tensor<512xf32>, %arg11: tensor<512xf32>, %arg12: tensor<512xf32>, %arg13: tensor<512xf32>, %arg14: tensor<512xf32>, %arg15: tensor<512xf32>, %arg16: tensor<512xf32>, %arg17: tensor<1x512xf32>, %arg18: tensor<512x512xf32>, %arg19: tensor<512x512xf32>, %arg20: tensor<512x512xf32>, %arg21: tensor<512x512xf32>, %arg22: tensor<1536x512xf32>, %arg23: tensor<1536x512xf32>, %arg24: tensor<512x1536xf32>, %arg25: tensor<512x512xf32>, %arg26: tensor<512x512xf32>, %arg27: tensor<512x512xf32>, %arg28: tensor<512x512xf32>, %arg29: tensor<1536x512xf32>, %arg30: tensor<1536x512xf32>, %arg31: tensor<512x1536xf32>, %arg32: tensor<512x512xf32>, %arg33: tensor<512x512xf32>, %arg34: tensor<512x512xf32>, %arg35: tensor<512x512xf32>, %arg36: tensor<1536x512xf32>, %arg3 |