Vectorization
PyTorch Developer Podcast
English - May 07, 2021 13:00 - 14 minutes - 13.4 MB - ★★★★★ - 35 ratingsTechnology deep learning machine learning pytorch Homepage Download Apple Podcasts Google Podcasts Overcast Castro Pocket Casts RSS feed
Previous Episode: Dynamic library structure
Next Episode: Inference mode
What is vectorization? How do you use it in PyTorch? What are some of the traps and pitfalls of writing vectorized code in PyTorch?
Further reading:
native/cpu README https://github.com/.../aten/src/ATen/native/cpu/README.mdVec256 classes https://github.com/.../tree/master/aten/src/ATen/cpu/vec256AVX512 support tracking issue https://github.com/pytorch/pytorch/issues/56187