| 23 | namespace backends { |
| 24 | |
| 25 | std::tuple<ir::Module, ir::Module> SplitDeviceAndHostModule(ir::Module module) { |
| 26 | detail::CollectBucketStrategyHostFunctionVisitor visitor(module->name); |
| 27 | return visitor(module); |
| 28 | } |
| 29 | |
| 30 | ir::Module CreateSwitchWithBroadcastConditionModule( |
| 31 | const std::vector<ir::Expr> &broadcast_conditions, |