viable/strict/1789185010: add kernel shared memory get_graph_data() (#196527)
- PyTorch 近 90 天出现 833 次
- PyTorch 近 90 天第 822 次发版
- 上一次：同一天稍早 · viable/strict/1789170654: [MPS] Migrate softmax to metal kernels (#196071)
发生了什么
add kernel shared memory get_graph_data() shared_mem_bytes is the launch's dynamic shared memory and comes from the node params, right next to the dimensions already read there. It stays best-effort like the kernel-name lookup beside it: a failed params query reports None rather than raising, since a dump is a diagnostic and a missing number should not cost the caller the whole graph. Test Plan: LD_LIBRARY_PATH=/usr…
摘要按规则整理自下方来源原文