pytorch icon indicating copy to clipboard operation
pytorch copied to clipboard

import pytorch with rocm support will just crash on a CPU without AVX

Open echoidcf opened this issue 1 year ago • 0 comments

🐛 Describe the bug

rocm (rocrand or rocblas or something like that) defaultly use AVX instructions. When they are loaded by torch with rocm support, just import torch will crash everything. Because it will call a asm instruction called vxorps which is AVX, but if the CPU does not support AVX for example G4930 , BOOM!

I change my CPU from G4930 to i3 9100, everything goes well after that.

Versions

As far as test, rocm 5.4.x and pytorch 2.0.0 at least has this problem.

echoidcf avatar Apr 13 '23 14:04 echoidcf