test_topi_transform.py
32.3 KB
-
[codegen] Add multiple operands and function support when using fp16 compilation (#4056) · ce72e9b5* overload half operators for cuda codegen * add float16 te test_op_level1 * fix test_op_level1.py * fix lint * disable fp16 test if gpu does not support * disable fp16 test if gpu does not support * bypass float16 test if gpu does not support float16
Xingyu Zhou committed