Support for CUDA 11.2?

[TearGosling]

Hi,
I've been trying to use ColossalAI on Google Colab as a testing ground for future AI runs. Specifically, I want to get the FP16 optimizer working during training. However, Colab's T4 GPUs have a CUDA version of 11.2, which is a CUDA version that is not offered on the downloads page. I've tried to build ColossalAI from source, but I get an error that it failed to build the wheel for ColossalAI, and it does not install ColossalAI properly. Is it possible for CUDA 11.2 support to be added in the future? Thanks.