frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
9fff8155c362da777e7ce31b85fb2dc7cfced2d5
pytorch/torch/csrc/inductor/aoti_include
History
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
..
array_ref.h
cpp_wrapper: Move #includes to per-device header files (#145932)
2025-01-29 21:08:45 +00:00
common.h
[AOTI][reland2] Remove typedef for half and bfloat16 (#153467)
2025-05-14 02:37:18 +00:00
cpu.h
cpp_wrapper: Move #includes to per-device header files (#145932)
2025-01-29 21:08:45 +00:00
cuda.h
cpp_wrapper: Move #includes to per-device header files (#145932)
2025-01-29 21:08:45 +00:00
mps.h
[aoti] Add MPS runner and shim (#153964)
2025-05-23 05:45:35 +00:00
xpu.h
cpp_wrapper: Move #includes to per-device header files (#145932)
2025-01-29 21:08:45 +00:00