I have fixed an error when compiled with float16 model in cuda.
Related question in TVM Discuss can be found here.
Can someone check my pull request?
#3811
Here is the error detail.
/tmp/tmpz_0pydlm/my_kernel.cu(9890): error: more than one instance of overloaded function "max" matches the argument list:
function "max(int, int)"
function "max(unsigned int, unsigned int)"
function "max(int, unsigned int)"
function "max(unsigned int, int)"
function "max(long, long)"
function "max(unsigned long, unsigned long)"
function "max(long, unsigned long)"
function "max(unsigned long, long)"
function "max(long long, long long)"
function "max(unsigned long long, unsigned long long)"
function "max(long long, unsigned long long)"
function "max(unsigned long long, long long)"
function "max(float, float)"
argument types are: (half, __half)