Added changes to add --build_variant flag for cpu only build.

Author: Sunidhi-Gaonkar1

build.py

@@ -1239,12 +1239,7 @@ def create_dockerfile_linux(
 
 WORKDIR /opt/tritonserver
 COPY --chown=1000:1000 NVIDIA_Deep_Learning_Container_License.pdf .
-RUN if [ "$(uname -m)" = "ppc64le" ]; then \
-        VARIANT="cpu"; \
-    else \
-        VARIANT="all"; \
-    fi &&\
-    find /opt/tritonserver/python -maxdepth 1 -type f -name \
+RUN find /opt/tritonserver/python -maxdepth 1 -type f -name \
     "tritonserver-*.whl" | xargs -I {} pip install --upgrade {}[$VARIANT] && \
     find /opt/tritonserver/python -maxdepth 1 -type f -name \
     "tritonfrontend-*.whl" | xargs -I {} pip install --upgrade {}[$VARIANT];
@@ -1300,6 +1295,7 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
     df = """
 ARG TRITON_VERSION
 ARG TRITON_CONTAINER_VERSION
+ARG VARIANT=all
 
 ENV TRITON_SERVER_VERSION ${TRITON_VERSION}
 ENV NVIDIA_TRITON_SERVER_VERSION ${TRITON_CONTAINER_VERSION}
@@ -1453,15 +1449,6 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
 """
         else:
             df += """
-RUN if [ "$(uname -m)" = "ppc64le" ]; then \
-        apt-get update \
-        && apt-get install -y --no-install-recommends \
-        clang-15 \
-        && ln -s /usr/bin/clang-15 /usr/bin/clang -f \
-        && ln -s /usr/bin/clang++-15 /usr/bin/clang++ -f \
-        && pip3 install cython ninja; \
-    fi
-
 # python3, python3-pip and some pip installs required for the python backend
 RUN apt-get update \\
       && apt-get install -y --no-install-recommends \\
@@ -1913,6 +1900,10 @@ def create_docker_build_script(script_name, container_install_dir, container_ci_
                 f"--secret id=NVPL_SLIM_URL",
                 f"--build-arg BUILD_PUBLIC_VLLM={build_public_vllm}",
             ]
+        if  FLAGS.build_variant:
+            finalargs += [
+                "--build-arg VARIANT="+(FLAGS.build_variant),
+            ]
         finalargs += [
             "-t",
             "tritonserver",
@@ -2778,6 +2769,13 @@ def enable_all():
         default=DEFAULT_TRITON_VERSION_MAP["rhel_py_version"],
         help="This flag sets the Python version for RHEL platform of Triton Inference Server to be built. Default: the latest supported version.",
     )
+    parser.add_argument(
+        "--build_variant",
+        required=False,
+        type=str,
+        default="all",
+        help="Can be set to all or cpu,Default value is all."
+    )
     parser.add_argument(
         "--build-secret",
         action="append",
@@ -2817,6 +2815,9 @@ def enable_all():
         FLAGS.extra_backend_cmake_arg = []
     if FLAGS.build_secret is None:
         FLAGS.build_secret = []
+    if hasattr(FLAGS, 'build_variant') and FLAGS.build_variant not in ["all", "cpu"]:
+        raise ValueError(f"Invalid build_variant value: {FLAGS.build_variant}. Expected 'all' or 'cpu'.")
+
 
     # if --enable-all is specified, then update FLAGS to enable all
     # settings, backends, repo-agents, caches, file systems, endpoints, etc.