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 |
否 |