viable/strict/1790789324: Fix _int_mm CPU corner case (#194682)
- PyTorch 近 90 天出现 1694 次
- PyTorch 近 90 天第 1673 次发版
- 上一次:同一天稍早 · viable/strict/1790775140
发生了什么
This is to fix #195066 _mkldnn_gemm_i8i8i32_with_blas took leading dimensions from the input strides, but the stride of a size-1 dim is arbitrary and inductor emits 0 for it in some cases. On the other hand, oneDNN's row-major gemm requires lda >= K so it will silently reject the call, causing accuracy failures in workloads like int8 quantized LLM decoding. The PR fixes this corner case and also check this oneDNN ge…
摘要按规则整理自下方来源原文