tree 4523c8e1d2392a9a151d491fc0cc7d9a72705efd
parent 9c0d5db71f32f9b03fe95d440f7fe80342dc6bef
author fredwang <fredwang@bytedance.com> 1628470603 +0800
committer fredwang <fredwang@bytedance.com> 1628501403 +0800

Update setup.py for the nvcc compilation options for half float.

Remove compute_35 and compute_50 because
1. they do not support half float;
2. google colab's GPU has been updated from K80 (compute_35) to T4 (compute_75).