jax.profiler.save_device_memory_profile#

jax.profiler.save_device_memory_profile(filename, backend=None)[原始碼]#

收集裝置記憶體效能分析，並寫入檔案。

save_device_memory_profile() 是 device_memory_profile() 的便利包裝函式，可將其輸出儲存到 filename。 有關更多資訊，請參閱 device_memory_profile() 文件。

參數：

  • filename – 應寫入效能分析的檔案名稱。

  • backend (str | None | None) – (選填)；應收集裝置記憶體效能分析的 JAX 後端名稱。

回傳型別：

None