Ean Garvey 6a82667778 Add pretraining pytest script with SHARK dynamo.

2023-06-15 12:01:35 -05:00

..

examples

Update OPT, ResNet example scripts. (#1492)

2023-06-05 20:19:35 -07:00

iree_utils

Remove deprecared --iree-mhlo-demote-164-to-132 flag usage. (#1533)

2023-06-13 22:40:47 -05:00

sharkdynamo

Add pretraining pytest script with SHARK dynamo.

2023-06-15 12:01:35 -05:00

tests

Add stress test

2022-12-13 13:21:51 -08: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

[SharkInference] Make SharkInference compile the entire module (#708)

2023-01-03 23:25:23 +05:30

model_annotation.py

Modify the lowering config format for SPIRVMatmulPromoteVectorize pipeline (#1471)

2023-05-29 21:53:48 -07:00

parser.py

Separate pytest benchmark modes and fix model updates for SHARK downloader / pytest. (#1264)

2023-04-03 08:24:21 -07:00

shark_benchmark_runner.py

Separate pytest benchmark modes and fix model updates for SHARK downloader / pytest. (#1264)

2023-04-03 08:24:21 -07:00

shark_downloader.py

Revive SD downloads from shark_tank. (#1465)

2023-05-25 12:03:21 -05:00

shark_generate_model_config.py

Add a model config generator (#1511)

2023-06-09 15:32:00 -07:00

shark_importer.py

mega vicuna merge

2023-06-14 19:06:29 +05:30

shark_inference.py

Fix formatting issues. (#903)

2023-02-01 09:12:45 +05:30

shark_runner.py

Change mlir dialects for tf tests to stablehlo. (#1535)

2023-06-14 10:43:49 -07:00

shark_trainer.py

Change instances of input_type='mhlo' to 'auto' (#1482)

2023-06-02 16:43:47 -05:00

stress_test.py

Fix concurrency issue in stress_test for CUDA devices

2022-12-22 08:54:19 -08: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

Add mlir_type for torch_model_list.csv (#1428)

2023-05-15 10:23:54 -07:00