pytorch / FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is FBGEMM-GPU support aarch64 arch?

Qiyu8 opened this issue · comments

I want to build FBGEMM under ARM-based machine, but there is a compile error :
g++: error: unrecognized command line option '-mavx'
g++: error: unrecognized command line option '-mf16c'
g++: error: unrecognized command line option '-mfma'
g++: error: unrecognized command line option '-mavx2'
GPU code should not coupled with X86 arch, Is FBGEMM-GPU support aarch64 arch?

Hi @Qiyu8, thanks for pointing out this issue. Currently, FBGEMM and FBGEMM_GPU is built and tested for x86 architecture, and there is no official support for ARM yet. @shintaro-iwasaki would you happen to have insights into this effort?

Thank you for letting us know! We cannot give an exact ETA at this point, we are looking at this issue.