message: Keep small clusters together
diff: @@ -16,6 +16,7 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  +import os
  import numpy as np
  from tensorflow.compiler.plugin.poplar.ops import gen_ipu_ops
  @@ -232,4 +233,6 @@ class PopDatastreamTest(test_util.TensorFlowTestCase):
  if __name__ == "__main__":
  + os.environ['TF_XLA_FLAGS'] = ('--tf_xla_min_cluster_size=1 ' +
  + os.environ.get('TF_XLA_FLAGS', ''))
  googletest.main()
  