Daniel Garvey 75672c0e28 set task_topology_max_group to cpu_count (#1594) ...

by default. Can be overriden with a flag of the same str

2023-06-26 14:54:06 -07:00

..

dynamo_backend

Make the tensors contiguous.

2023-06-26 17:29:54 +05:30

examples

Add patch for making compile API work for both MEGABYTE and MiniGPT4 (#1559)

2023-06-20 10:04:17 -07:00

iree_utils

set task_topology_max_group to cpu_count (#1594)

2023-06-26 14:54:06 -07:00

shark_eager

Add shark_eager mode.

2023-06-22 22:59:14 +05:30

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

set task_topology_max_group to cpu_count (#1594)

2023-06-26 14:54:06 -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

downloader was double downloading (#1580)

2023-06-22 18:30:27 -07:00

shark_generate_model_config.py

Add dispatch-level config file generator for manual annotation (#1566)

2023-06-22 15:11:41 -07:00

shark_importer.py

Add shark_eager mode.

2023-06-22 22:59:14 +05:30

shark_inference.py

[SharkInference-SharkRuntime] Adds capability to mmap vmfbs

2023-06-22 20:43:40 +05:30

shark_runner.py

[SharkInference-SharkRuntime] Adds capability to mmap vmfbs

2023-06-22 20:43:40 +05:30

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