Previously we support a limited case of function type deduction and in many places we have to supply the type twice during set_body_typed (one in the template parameter, another in the lambda signature). This PR improves the deduce function by enablng automatic function signature deduction. ``` TVM_REGISTER_GLOBAL("sub") .set_body_typed([](int x, int y) -> int { return x - y; }); ``` Unfortunately, because of template conflict, we can not support the original case where both type signature and lambda are supplied through set_body_typed. This PR refactors the existing regsitration to the new style.
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
.gitignore | Loading commit data... | |
attrs_test.cc | Loading commit data... | |
build_module_test.cc | Loading commit data... | |
container_test.cc | Loading commit data... | |
expr_test.cc | Loading commit data... | |
ir_functor_test.cc | Loading commit data... | |
ir_simplify_test.cc | Loading commit data... | |
ir_ssa_test.cc | Loading commit data... | |
object_protocol_test.cc | Loading commit data... | |
packed_func_test.cc | Loading commit data... | |
pattern_match_test.cc | Loading commit data... | |
relay_build_module_test.cc | Loading commit data... | |
relay_pass_type_infer_test.cc | Loading commit data... | |
relay_transform_sequential.cc | Loading commit data... | |
simple_passes_test.cc | Loading commit data... | |
tensor_test.cc | Loading commit data... | |
threading_backend_test.cc | Loading commit data... | |
topi_ewise_test.cc | Loading commit data... | |
utvm_runtime_standalone_test.cc | Loading commit data... |