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
580 Listeners
Science Weekly
427 Listeners
Cautionary Tales with Tim Harford
5,146 Listeners
All-In with Chamath, Jason, Sacks & Friedberg
10,043 Listeners