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 伴隨問答
針對這篇報導有疑問?可直接點選快捷問題或向 AI 追問背景脈絡與技術細節:
信源出處github.com
trunk/42ab44e523b563210e07ca5bbb89d83bf69518cc: [MPS] Migrate isin to Metal (#202930)
本文為 888news 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。
來源: PyTorch Core Releases (github.com)
開啟原文