apache--tvm

Files

T

Chris Sullivan 1420df7744 [TE] Support schedulable TIR compute definitions in TOPI (#11589 )

This PR adds `te.extern_primfunc` which provides the interface around TE ExternOp that allows a TVMScript defined schedulable TIR PrimFunc to be inlined into a TE compute graph. The result is that TIR can be used for compute definitions in Relay OpStrategies and, paired with meta-scheduler support in relay as introduced in #10578, these compute definitions can be scheduled and tuned as demonstrated in the attached tests.  

Prior to this, compute definitions were limited to those definable in TE only. As a consequence of this patch and ongoing improvements to TVMScript meta-programming (#11097), TOPI can be extended to include compute and scheduling functions targeting schedulable TIR uniformly.

2022-06-13 08:50:19 -07:00

hybrid

[TIR] Fix int32 vs int64 mismatch in For construct. (#10595 )

2022-04-05 15:08:36 +09:00

__init__.py

[TE] Support schedulable TIR compute definitions in TOPI (#11589 )

2022-06-13 08:50:19 -07:00

_ffi_api.py

[REFACTOR][PY] Establish tvm.te and tvm.driver (#4900 )

2020-02-17 18:47:09 -08:00

autodiff.py

[TE] reverse-mode autodiff without any optimization (#5121 )