@Truong_Khang
Can you try
CAFFE2_PERF_WITH_AVX2 =1 python setup.py install ?
Not sure what is your env.
I recommend file an issue in github pytorch repo and label it build, we will have expert there helping you.