powderluv 128342f47f Update stable_diffusion_amd.md

2022-11-30 13:42:25 -08:00

..

examples

Update stable_diffusion_amd.md

2022-11-30 13:42:25 -08:00

iree_utils

dump all isas with amdllpc (#517)

2022-11-30 11:33:30 -08:00

sharkdynamo

Add sharkdynamo which combines shark with torchdynamo.

2022-09-23 22:40:02 +05:30

tests

Remove tf package dep for SharkDownloader tflite tests (#212)

2022-07-26 12:41:12 -07:00

__init__.py

Modify the directory structure to remove conflicts.

2022-04-04 21:23:33 +05:30

backward_makefx.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

iree_eager_backend.py

Fix IREE eager backend device string (#237)

2022-08-03 12:09:52 -07:00

model_annotation.py

Update model annotation to take vulkan configs (#495)

2022-11-17 14:34:17 -08:00

parser.py

Added a dispatch benchmarking tool (#441)

2022-10-28 14:31:03 -07:00

shark_benchmark_runner.py

Serialize torch-mlir CAPI module as bytecode instead of string. (#435)

2022-10-27 14:37:15 -05:00

shark_downloader.py

Add internet connection check for re-downloading models (#488)

2022-11-14 13:56:42 -06:00

shark_importer.py

Changes incorporating the recent torch_mlir compile api changes.

2022-11-15 15:25:37 +05:30

shark_inference.py

dump spv for dispatches (#509)

2022-11-23 22:34:27 -06:00

shark_runner.py

Actually print the error message when SharkRunner can't initialize the driver (#482)

2022-11-13 19:08:46 -08:00

shark_trainer.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

tflite_utils.py

Remove tf package dep for SharkDownloader tflite tests (#212)

2022-07-26 12:41:12 -07:00

torch_mlir_lockstep_tensor.py

[Lockstep] Hack to avoid aten._reshape_alias (#332)

2022-09-22 18:02:09 -04:00

torch_mlir_utils.py

Changes incorporating the recent torch_mlir compile api changes.

2022-11-15 15:25:37 +05:30