You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
PyTorch Channels Last memory format perf optimization and oneDNN integration plan.
PyTorch Channels Last Memory Format Performance Optimization on CPU Path
("mkldnn" has been renamed to "oneDNN", but exsiting PyTorch APIs still use "mkldnn", future work will align PyTorch user level APIs to "oneDNN")
Table of Contents
PyTorch Channels Last memory format introduction
oneDNN API for NHWC layout
Generic Channels Last memory format optimization with ATen native
oneDNN NHWC integration
NB: Memory format refers to data representation that describes how multidimensional arrays (nD) are stored in linear (1D) memory address space. Memory format has the same semantic with layout in oneDNN. Layout in PyTorch has other semantic ofdescribing dense or sparse with the attributes: 'torch.strided', 'torch.sparse_coo'.