CMake Error at caffe2_gpu_generated_THCBlas.cu.o.Release.cmake

python setup.py install
How to solve the following error?

/pytorch/aten/src/ATen/core/Half-inl.h(21): error:identifier “__half_as_short” is undefined
1 error detected in the compilation of “/tmp/tmpxft_000087bd_00000000-7_THCBlas.cpp1.ii”.
CMake Error at caffe2_gpu_generated_THCBlas.cu.o.Release.cmake:279(message):

I meeted the same problem with you~