PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
Strides
20 minutes Posted May 30, 2022 at 4:00 pm.
0:00
20:31
Download MP3
Show notes

Mike Ruberry has an RFC about stride-agnostic operator semantics (https://github.com/pytorch/pytorch/issues/78050), so let's talk about strides. What are they? How are they used to implement views and memory format? How do you handle them properly when writing kernels? In what sense are strides overspecified, and therefore, not worth slavishly reimplementing in a system like PrimTorch? What does Edward think we should do about them?

My blog post that covers strides along with other topics can be found at http://blog.ezyang.com/2019/05/pytorch-internals/