Ramin Azarmehr cdfd0ea162 [MPS] Introduce torch.mps.Event() APIs (#102121) ... - Implement `MPSEventPool` to recycle events. - Implement python bindings with `torch.mps.Event` class using the MPSEventPool backend. The current member functions of the Event class are `record()`, `wait()`, `synchronize()`, `query()`, and `elapsed_time()`. - Add API to measure elapsed time between two event recordings. - Added documentation for Event class to `mps.rst`. - Added test case to `test_mps.py`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/102121 Approved by: https://github.com/albanD, https://github.com/kulinseth		2023-08-08 03:45:45 +00:00
..
__init__.py	[MPS] Introduce torch.mps.Event() APIs (#102121)	2023-08-08 03:45:45 +00:00
event.py	[MPS] Introduce torch.mps.Event() APIs (#102121)	2023-08-08 03:45:45 +00:00
profiler.py	Apply UFMT to low traffic torch modules (#106249)	2023-07-29 23:37:30 +00:00