(self, mod: IRModule, ctx: PassContext)
| 85 | """Pass to dispatch scan and sort operators to platform dependent implementation.""" |
| 86 | |
| 87 | def transform_module(self, mod: IRModule, ctx: PassContext) -> IRModule: |
| 88 | sampling_dispatcher = SamplingDispatcher(mod) |
| 89 | for gv, func in mod.functions_items(): |
| 90 | if isinstance(func, relax.Function): |
| 91 | func = sampling_dispatcher.visit_expr(func) |
| 92 | sampling_dispatcher.builder_.update_func(gv, func) |
| 93 | return sampling_dispatcher.builder_.finalize() |
nothing calls this directly
no test coverage detected