message: update distributed cmake flags
diff: @@ -419,6 +419,7 @@ def main():
  parser.add_argument(
  '--distributed_build',
  + type=str,
  help="Builds a distributed version of the nGraph components\n",
  action="store")
  @@ -520,10 +521,12 @@ def main():
  if arguments.debug_build:
  ngraph_cmake_flags.extend(["-DCMAKE_BUILD_TYPE=Debug"])
  - if arguments.distributed_build:
  - ngraph_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_OMPI_ENABLE=TRUE"])
  + if (arguments.distributed_build=="OMPI"):
  + ngraph_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_ENABLE=OMPI"])
  + elif (arguments.distributed_build=="MLSL"):
  + ngraph_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_ENABLE=MLSL"])
  else:
  - ngraph_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_OMPI_ENABLE=FALSE"])
  + ngraph_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_ENABLE=OFF"])
  if arguments.build_gpu_backend:
  ngraph_cmake_flags.extend(["-DNGRAPH_GPU_ENABLE=YES"])
  @@ -563,7 +566,7 @@ def main():
  ngraph_tf_cmake_flags.extend(["-DUNIT_TEST_TF_CC_DIR=" +
  os.path.join(artifacts_location, "tensorflow")])
  - if (arguments.distributed_build):
  + if ((arguments.distributed_build=="OMPI") or (arguments.distributed_build=="MLSL")):
  ngraph_tf_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_ENABLE=TRUE"])
  else:
  ngraph_tf_cmake_flags.extend(["-DNGRAPH_DISTRIBUTED_ENABLE=FALSE"])
  