TestJit.test_fusion_distribute_cuda.expect

graph(%0 : Float(*, *)
      %1 : Float(*, *)) {
  %2 : Dynamic[] = prim::ListConstruct(%0, %1)
  %3 : Dynamic[] = aten::broadcast_tensors(%2)
  %4 : Dynamic, %5 : Dynamic = prim::ListUnpack(%3)
  %6 : Float(*, *) = prim::FusionGroup_0[device=0](%5, %4)
  return (%6);
}
with prim::FusionGroup_0 = graph(%0 : Dynamic
      %1 : Dynamic) {
  %2 : Float(*, *), %3 : Float(*, *) = prim::ConstantChunk[chunks=2, dim=1](%1)
  %4 : Float(*, *), %5 : Float(*, *) = prim::ConstantChunk[chunks=2, dim=1](%0)
  %6 : int = prim::Constant[value=1]()
  %7 : Float(*, *) = aten::add(%2, %4, %6)
  %8 : int = prim::Constant[value=1]()
  %9 : Float(*, *) = aten::add(%3, %5, %8)
  %10 : Float(*, *) = aten::mul(%7, %9)
  return (%10);
}