Solved by Vasiliy_Kuznetsov in post #2 
              
              
                Hi @saykarthik, please feel free to take a look at Add AVX512 support in ATen & remove AVX support by imaginary-person · Pull Request #56992 · pytorch/pytorch · GitHub which is in progress and which aims to add this support.  There are performance numbers planned to be added the PR before it’s ready…