Ean Garvey 9c8cbaf498 Add support for ROCM (Windows) in Studio + compile utils (#1770) ...

* WIP: MSVC ROCM support for SHARK Studio

* Make get_iree_rocm_args platform-agnostic.

* Update stable_args.py

* Update rocm arg handling in SD utils

* Guard quantization imports.

Co-authored-by: jam https://github.com/jammm

2023-08-25 20:56:05 -07:00

..

__init__.py

Divide iree_utils and do module imports on function calls.

2022-06-22 14:17:33 +05:30

_common.py

Add support for ROCM (Windows) in Studio + compile utils (#1770)

2023-08-25 20:56:05 -07:00

benchmark_utils.py

Add support for ROCM (Windows) in Studio + compile utils (#1770)

2023-08-25 20:56:05 -07:00

compile_utils.py

Add support for ROCM (Windows) in Studio + compile utils (#1770)

2023-08-25 20:56:05 -07:00

cpu_utils.py

Optimize device enumeration overhead and log details on long operations. (#1734)

2023-08-07 17:20:53 -07:00

gpu_utils.py

Add support for ROCM (Windows) in Studio + compile utils (#1770)

2023-08-25 20:56:05 -07:00

metal_utils.py

Optimize device enumeration overhead and log details on long operations. (#1734)

2023-08-07 17:20:53 -07:00

trace.py

Optimize device enumeration overhead and log details on long operations. (#1734)

2023-08-07 17:20:53 -07:00

vulkan_target_env_utils.py

Optimize device enumeration overhead and log details on long operations. (#1734)

2023-08-07 17:20:53 -07:00

vulkan_utils.py

Optimize device enumeration overhead and log details on long operations. (#1734)

2023-08-07 17:20:53 -07:00