where we combine the shape and dtype parameters into out_sinfo. In the concrete CallNode of call_tir, the output shape is no longer passed as an CallNode argument and passed in the sinfo_args instead.
For call_packed, call_builtin and invoke_closure, we change the type_args parameter to sinfo_args. For example, the API of call_packed is updated to
Since we changed the API of call_tir - the CallNode does not contain the shape as one argument, - together with the fact that the dataflow pattern language does not yet support matching StructInfo, we have no approach to match the shape anymore. Therefore, the shape parameter is removed from is_call_tir. I left some Todo items there for future sinfo_args matching support.
cc @Hzfengsy @tqchen @slyubomirsky @spectrometerHBH
Following #377 and #379, this PR followups to update the API of
call_tir
,call_packed
,call_builtin
andinvoke_closure
.call_tir
is changed towhere we combine the
shape
anddtype
parameters intoout_sinfo
. In the concrete CallNode ofcall_tir
, the output shape is no longer passed as an CallNode argument and passed in thesinfo_args
instead.call_packed
,call_builtin
andinvoke_closure
, we change thetype_args
parameter tosinfo_args
. For example, the API ofcall_packed
is updated toOne thing specific to note is about our existing dataflow pattern language. Previously we have a sugar function for
call_tir
:Since we changed the API of
call_tir
- the CallNode does not contain the shape as one argument, - together with the fact that the dataflow pattern language does not yet support matching StructInfo, we have no approach to match the shape anymore. Therefore, theshape
parameter is removed fromis_call_tir
. I left some Todo items there for futuresinfo_args
matching support.cc @Hzfengsy @tqchen @slyubomirsky @spectrometerHBH