trunk/e320689d0698ae12305459e36b87ab5a0c64fccc: [FakeTensor] Support sparse compressed tensor ctors (#192091)
- PyTorch 近 90 天出现 969 次
- PyTorch 近 90 天第 958 次发版
- 上一次:同一天稍早 · trunk/e42310637e81856ed12083d1aacf6fa229f6bcab: [xpu] Refine and improve MemPool functionality (#192048)
发生了什么
Fake impls for aten.sparse_compressed_tensor overloads run the meta kernel and restore the output device, instead of failing with a device mismatch. _to_dense fake impl now covers all sparse layouts, not just COO. FunctionalTensor wrapping a sparse tensor answers C++ layout() queries (dispatch_layout + prim.layout metadata fn); previously this hard-aborted via TensorImpl::layout_impl. Real-tensor propagation and FX…
摘要按规则整理自下方来源原文