angelayi b33b7d5c8c [aoti] Add MPS runner and shim (#153964) ...

Added AOTIModelContainerRunnerMps and a shim for mps fallback ops.
I also added a mps-specific shim which contains one operator, which will be used to set arguments being passed to the Metal kernel:

```
AOTI_TORCH_EXPORT AOTITorchError aoti_torch_mps_set_arg(
    AOTIMetalKernelFunctionHandle func,
    unsigned idx,
    AtenTensorHandle tensor);
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153964
Approved by: https://github.com/malfet, https://github.com/desertfire

2025-05-23 05:45:35 +00:00

..

_autoheuristic

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

aoti

[AOTI] Add _weight_int4pack_mm to the C shim fallback list (#151059)

2025-04-11 21:22:35 +00:00

api

[BE] Ensure generated stub files by gen_pyi are properly formatted (#150730)

2025-05-17 12:30:40 +00:00

decompositions

Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"

2025-01-04 14:17:20 +00:00

dest

[torchgen] Refactor torchgen.utils.FileManager to accept pathlib.Path (#150726)

2025-05-15 02:52:24 +00:00

executorch

[torchgen] Refactor torchgen.utils.FileManager to accept pathlib.Path (#150726)

2025-05-15 02:52:24 +00:00

fuse

[BE] update type annotations for basic utilities in torch/__init__.py (#129001)

2024-06-24 18:04:38 +00:00

operator_versions

[3/N] Use internal linkage in C++ files (#151297)

2025-05-05 17:48:39 +00:00

selective_build

[BE][Easy] enable postponed annotations in torchgen (#129376)

2024-06-29 09:23:39 +00:00

shape_functions

Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"

2025-01-04 14:17:20 +00:00

static_runtime

[1/N] Use std::string_view in torchgen (#146403)

2025-04-16 01:50:22 +00:00

__init__.py

Package config/template files with torchgen (#78942)

2022-06-07 13:33:55 +00:00

BUCK.oss

rename BUILD.buck to BUCK.oss (#78792)

2022-06-03 07:23:16 +00:00

BUILD.bazel

Rename tools/codegen to torchgen (#76275)

2022-04-25 01:38:06 +00:00

build.bzl

update rules_python and let bazel install its own pip dependencies (#101405)

2023-05-23 06:20:33 +00:00

code_template.py

[BE] Ensure generated stub files by gen_pyi are properly formatted (#150730)

2025-05-17 12:30:40 +00:00

context.py

[2/N] Apply py39 ruff fixes (#141938)

2024-12-05 06:26:06 +00:00

gen_aoti_c_shim.py

[1/N] Use std::string_view in torchgen (#146403)

2025-04-16 01:50:22 +00:00

gen_backend_stubs.py

[BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546)

2025-02-27 20:46:16 +00:00

gen_executorch.py

[BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546)

2025-02-27 20:46:16 +00:00

gen_functionalization_type.py

Add env for disabling meta reference on functionalization. (#148822)

2025-03-11 16:13:35 +00:00

gen_lazy_tensor.py

[BE][Easy] use pathlib.Path instead of dirname / ".." / pardir (#129374)

2024-12-29 17:23:13 +00:00

gen_schema_utils.py

[2/N] Apply py39 ruff fixes (#141938)

2024-12-05 06:26:06 +00:00

gen_vmap_plumbing.py

[BE][CI] bump ruff to 0.9.0: string quote styles (#144569)

2025-02-24 19:56:09 +00:00

gen.py

[aoti] Add MPS runner and shim (#153964)

2025-05-23 05:45:35 +00:00

local.py

[BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546)

2025-02-27 20:46:16 +00:00

model.py

[MTIA ATen Backend] Migrate "_unsafe_view" and "view" ops from out-of-tree to pytorch in-tree (#153670)

2025-05-21 05:20:45 +00:00

native_function_generation.py

[aotd] capture rrelu_with_noise noise mutation in compile (#141867)

2024-12-04 12:18:58 +00:00

utils.py

[torchgen] Refactor torchgen.utils.FileManager to accept pathlib.Path (#150726)

2025-05-15 02:52:24 +00:00

yaml_utils.py

[BE][CI] bump ruff to 0.9.2: multiline assert statements (#144546)

2025-02-27 20:46:16 +00:00