Sign up to save your podcasts
Or
Share Vectorization
Share to email
Share to Facebook
Share to X
Share to email
Share to Facebook
Share to X
Or
Vectorization
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.md
- Vec256 classes https://github.com/.../tree/master/aten/src/ATen/cpu/vec256
- AVX512 support tracking issue https://github.com/pytorch/pytorch/issues/56187
View all episodes
By Edward Yang, Team PyTorch
4.8
4949 ratings
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.md
- Vec256 classes https://github.com/.../tree/master/aten/src/ATen/cpu/vec256
- AVX512 support tracking issue https://github.com/pytorch/pytorch/issues/56187
More shows like PyTorch Developer Podcast
View all
Talk Python To Me
585 Listeners
Science Weekly
416 Listeners
Cautionary Tales with Tim Harford
5,179 Listeners
All-In with Chamath, Jason, Sacks & Friedberg
9,901 Listeners