若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
---|---|---|
torch.mps.synchronize |
||
torch.mps.get_rng_state |
||
torch.mps.set_rng_state |
||
torch.mps.manual_seed |
||
torch.mps.seed |
||
torch.mps.empty_cache |
||
torch.mps.set_per_process_memory_fraction |
||
torch.mps.current_allocated_memory |
||
torch.mps.driver_allocated_memory |
||
torch.mps.profiler.start |
||
torch.mps.profiler.stop |
||
torch.mps.profiler.profile |
||
torch.mps.event.Event |
||
torch.mps.event.Event.elapsed_time |
||
torch.mps.event.Event.query |
||
torch.mps.event.Event.record |
||
torch.mps.event.Event.synchronize |
||
torch.mps.event.Event.wait |