commit | 3ca846e3dfbbf50ae70034083206e72de860cb62 | [log] [tgz] |
---|---|---|
author | fredwang <fredwang@bytedance.com> | Mon Aug 09 08:56:43 2021 +0800 |
committer | fredwang <fredwang@bytedance.com> | Mon Aug 09 17:30:03 2021 +0800 |
tree | 4523c8e1d2392a9a151d491fc0cc7d9a72705efd | |
parent | 9c0d5db71f32f9b03fe95d440f7fe80342dc6bef [diff] |
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).