void fuseBroadcast(std::shared_ptr<Graph>& graph) {
for(auto n : graph->nodes()) {
// Can't fuse into nodes that don't support broadcasting
if (!isBroadcasting(n)) continue;
// If the node already broadcasts, can't "rebroadcast"
// TODO: Actually, maybe you can, if there is a broadcast for some
// dims, and then another broadcast for the rest. But this will
// never happen in practice so I didn't implement it.
if (n->hasAttribute(kbroadcast) && n->i(kbroadcast)) continue;
JIT_ASSERT(!n->hasAttribute(kaxis));
auto input_index = n->inputs().size() - 1;
auto* expanded_rhs = n->input(input_index)->node();
// The expanded_rhs input isn't actually an expand, so no fusion available
if (expanded_rhs->kind() != kExpand) continue;
auto* unexpanded_rhs = expanded_rhs->input();
// We need to know what the type pre-expand is. We should basically
// always have this information (because expands are only ever traced,
// not generated from symbolic), but if for some reason we don't
// have it, we need to skip.
if (!unexpanded_rhs->isTensor()) continue;
// Not all broadcasts are supported by ONNX broadcast.
if (!fusibleExpandTo(unexpanded_rhs->type()->expect<TensorType>()->sizes(), // from
expanded_rhs->output()->type()->expect<TensorType>()->sizes()) // to
) continue;
n->replaceInput(input_index, unexpanded_rhs);
n->i_(kbroadcast, 1);
if (!expanded_rhs->hasUses()) {
expanded_rhs->destroy();
}
}
}
void DvbStream::stopBroadcast()
{
int i;
DVBout *o=0;
QPtrList<DVBout> p;
for ( i=0; i<(int)out.count(); i++ ) {
o = out.at(i);
o->stopBroadcast();
if ( !o->hasLive() && !o->hasRec() )
p.append( o );
}
for ( i=0; i<(int)p.count(); i++ ) {
removePids( p.at(i) );
if ( cam )
cam->stopService( &(p.at(i)->channel) );
removeOut( p.at(i) );
}
if ( out.count()==0 )
stop();
emit isBroadcasting( false );
}
