Add New Operator

Hidet is designed to be extensible. It is easy to add new operators to Hidet. There are two ways to add and schedule an operator.

1. Rule-based Scheduling Define the mathematical computation of the operator, and Hidet will automatically schedule the computation into parallel tensor program with Hidet’s rule-based scheduler.

2. Template-based Scheduling Besides the computation, user can also give the concrete implementation of the operator to achieve better performance for complex operators.

Define Computation

• Define Operator Computation

Two Scheduling Methods

• Using Rule-based Scheduling
• Using Template-based Scheduling