PyTorch MKLDNN zero_() can overflow half-precision CPU tensors
A reported CPU path writes float32-sized zeros into bfloat16 and float16 MKLDNN buffers, crashing the process.
By tensorA reported CPU path writes float32-sized zeros into bfloat16 and float16 MKLDNN buffers, crashing the process.
By tensorCPU and other compiled fusion kernels cannot sit in an FX graph today, so the fx_wrapper path fails outside Triton.
By tensorNon-contiguous batch dimensions on the out tensor make the CPU path step wrong and can corrupt the heap; CUDA is fine.
By tensorA proposed change would process output channels concurrently and cut redundant writes on common Fold shapes.
By tensorThe CPU path indexes running mean and variance by channel count without checking buffer length, causing heap out-of-bounds access.
By tensorUnchecked offsets in _reinterpret_tensor let callers produce views that read past storage and trigger heap buffer overflows.
By tensorFull-width vector stores on tail blocks of 2D-tiled reductions corrupt memory or silently rewrite neighboring output rows.
By tensor