TestFuser.test_chunk_distributes_cuda.expect

graph(%x : Float(*, *),
      %y : Float(*, *)):
  %2 : Tensor[] = prim::ListConstruct(%x, %y)
  %3 : Tensor[] = aten::broadcast_tensors(%2)
  %4 : Tensor, %5 : Tensor = prim::ListUnpack(%3)
  %6 : Float(*, *) = prim::FusionGroup_0(%5, %4)
  return (%6)
with prim::FusionGroup_0 = graph(%0 : Tensor,
      %1 : Tensor):
  %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 : Float(*, *) = aten::add(%3, %5, %6)
  %9 : Float(*, *) = aten::mul(%7, %8)
  return (%9)