trunk/cf471dc229ebb2d98793f00f3150c370872b6cb8: [BE][MPS] Migrate remaining logical ops to Metal (#195592)
- PyTorch 近 90 天出现 402 次
- PyTorch 近 90 天第 399 次发版
- 上一次：同一天稍早 · ciflow/torchtitan/195649: [mtia_compiler][Pytorch] Support stream APIs in torch.compile
发生了什么
Using c10:metal::cast_to primitive to have consistent behavior for complex numbers, which fixes bug in MPSGraph implementation when op was called for complex types (previously only real part were checked, not complex value is truthy when either real or complex parts are nonzero) Pull Request resolved: #195592 Approved by: https://github.com/Skylion007
摘要按规则整理自下方来源原文