pytorch/caffe2
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
..
core Record the XPU and XCCL build settings in the compiled binary (#147161) 2025-05-20 09:21:39 +00:00
perfkernels Optimize SVE embedding performance (#150176) 2025-04-07 18:01:54 +00:00
serialize Update PyTorchStreamReader API to take cpu allocator override (#150439) 2025-04-18 01:53:14 +00:00
utils [3/N] Use internal linkage in C++ files (#151297) 2025-05-05 17:48:39 +00:00
.clang-format [BE][clang-format] make macro PyObject_HEAD_INIT(type) and PyVarObject_HEAD_INIT(type, size) have its own line (#136949) 2024-10-02 18:39:22 +00:00
CMakeLists.txt [aoti] Add MPS runner and shim (#153964) 2025-05-23 05:45:35 +00:00
unexported_symbols.lds
version_script.lds