Fix aarch64 cpu/cuda include logic#6584

atalman merged 1 commit intopytorch:mainpytorch/test-infra:mainfrom

atalman:fix_generate_matrixatalman/test-infra:fix_generate_matrixCopy head branch name to clipboard