Hi everyone,
I’d like to use one of my TIR passes as scheduling primitive, i.e.
s = tir.Schedule(my_primfunc)
s.my_custom_scheduling_primitive(y, params)
Is there a way of registering a tvm.tir.transform.PrimFuncPass
, a tvm.transform.module_pass
or a tvm.tir.stmt_functor.ir_transform
to the Schedule?
Thanks, Michael