Prashant Kumar pashu123

😇

Working from home

Compilers @amd

pashu123 / stable_diff_f16_elided.mlir

Created October 14, 2022 14:16

This file has been truncated, but you can view the full file.

pashu123 / resnet_50_fp16.py

Created October 13, 2022 17:50

	import numpy as np
	from shark.shark_inference import SharkInference
	from shark.shark_importer import SharkImporter
	from shark.shark_downloader import download_torch_model

	mlir_model, func_name, inputs, golden_out = download_torch_model(
	"resnet_50_fp16_old", tank_url="gs://shark_tank/prashant_nod"
	)

	shark_module = SharkInference(mlir_model, func_name, mlir_dialect="linalg")

pashu123 / resnet_50_fp16.py

Last active October 13, 2022 17:42

	import numpy as np
	from shark.shark_inference import SharkInference
	from shark.shark_importer import SharkImporter
	from shark.shark_downloader import download_torch_model

	mlir_model, func_name, inputs, golden_out = download_torch_model(
	"resnet_50_fp16_torch", tank_url="gs://shark_tank/prashant_nod"
	)

	shark_module = SharkInference(mlir_model, func_name, mlir_dialect="linalg")

pashu123 / elided.mlir

Created October 11, 2022 17:32

This file has been truncated, but you can view the full file.

pashu123 / unet_script.py

Created October 11, 2022 13:21

	import numpy as np
	from shark.shark_inference import SharkInference
	from shark.shark_importer import SharkImporter
	from shark.shark_downloader import download_torch_model

	mlir_model, func_name, inputs, golden_out = download_torch_model(
	"stable_diff_quant", tank_url="gs://shark_tank/prashant_nod"
	)

	shark_module = SharkInference(mlir_model, func_name, mlir_dialect="linalg", device="vulkan")

pashu123 / unet_script.py

Created October 11, 2022 13:04

	import torch
	import numpy as np
	from shark.shark_inference import SharkInference
	from shark.shark_importer import SharkImporter
	from shark.shark_downloader import download_torch_model

	mlir_model, func_name, inputs, golden_out = download_torch_model(
	"stable_diff_quant", tank_url="gs://shark_tank/prashant_nod"
	)

pashu123 / _lambda.mlir

Created October 10, 2022 18:04

This file has been truncated, but you can view the full file.

	#loc0 = loc(unknown)
	module attributes {torch.debug_module_name = "_lambda"} {
	func.func private @__torch__.torch.fx.graph_module._lambda.forward(%arg0: !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> loc(unknown), %arg1: !torch.tensor {torch.type_bound = !torch.vtensor<[2,4,64,64],f32>} loc(unknown), %arg2: !torch.tensor {torch.type_bound = !torch.vtensor<[],si64>} loc(unknown), %arg3: !torch.tensor {torch.type_bound = !torch.vtensor<[2,77,768],f32>} loc(unknown)) -> !torch.tensor {
	%3919 = torch.tensor_static_info_cast %arg1 : !torch.tensor to !torch.tensor<[2,4,64,64],f32> loc(#loc0)
	%3920 = torch.tensor_static_info_cast %arg2 : !torch.tensor to !torch.tensor<[],si64> loc(#loc0)
	%3921 = torch.tensor_static_info_cast %arg3 : !torch.tensor to !torch.tensor<[2,77,768],f32> loc(#loc0)
	%3922 = torch.prim.GetAttr %arg0["_param_constant365"] : !torch.nn.Module<"__torch__.torch.fx.graph_module._lambda"> -> !torch.tensor loc(#loc0)
	%3923 = torch.prim.GetAttr %arg0["_param_constant3

pashu123 / tf_unet.py

Created October 6, 2022 13:29

	import argparse
	from shark.shark_inference import SharkInference
	import numpy as np

	p = argparse.ArgumentParser(
	description=__doc__, formatter_class=argparse.ArgumentDefaultsHelpFormatter
	)

	p.add_argument("--mlir_loc", type=str, default=None, help="the device to use")
	args = p.parse_args()

pashu123 / tf_unet.py

Created October 5, 2022 16:42


	from iree import runtime as ireert
	from iree.compiler import tf as tfc
	from iree.compiler import compile_str
	import sys
	from absl import app

	import numpy as np
	import os
	import tempfile

pashu123 / tf_unet.py

Created October 5, 2022 14:02

	from stable_diffusion_tf.stable_diffusion import get_models

	from iree import runtime as ireert
	from iree.compiler import tf as tfc
	from iree.compiler import compile_str
	import sys
	from absl import app

	import numpy as np
	import os