跳到正文
PyTorch Core Releases(github.com)· Isalia20·· 3 小時前品質評分61

PyTorch MPS:將 isin 從 MPSGraph 遷移至 Metal

trunk/42ab44e523b563210e07ca5bbb89d83bf69518cc: [MPS] Migrate isin to Metal (#202930)

重點導讀

PyTorch 在最新的 trunk 版本中將 isin 從 MPSGraph 遷移至 Metal,大幅提升搜尋效能。

  • 速度提升:在 Apple M5 Pro 上,對 65536 個元素的搜尋,Metal 版比 MPSGraph 高達 88 倍。

AI 伴隨問答

自備 Key 深度分析

針對這篇報導有疑問?可直接點選快捷問題或向 AI 追問背景脈絡與技術細節:

信源出處github.com

trunk/42ab44e523b563210e07ca5bbb89d83bf69518cc: [MPS] Migrate isin to Metal (#202930)

本文為 888news 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。

開啟原文
來源: PyTorch Core Releases (github.com)
開啟原文

讀者討論

理性交流 · 即時討論