tvm_tirx.support.nvcc.have_fp16

Contents

tvm_tirx.support.nvcc.have_fp16#

tvm_tirx.support.nvcc.have_fp16(compute_version)#

Either fp16 support is provided in the compute capability or not

Parameters:

compute_version (str) – compute capability of a GPU (e.g. “6.0”)