[CI][Docs] Update nightly CPU package (#2903)

This PR updates the uses of the nightly CPU package

ci/task/pylint.sh

@@ -9,7 +9,7 @@ export PYTHONPATH="./python":${PYTHONPATH:-""}
 if [[ -n ${MLC_CI_SETUP_DEPS:-} ]]; then
     echo "MLC_CI_SETUP_DEPS=1 start setup deps"
     # TVM Unity is a dependency to this testing
-    pip install --quiet --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+    pip install --quiet --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
     pip install --quiet --pre -U cuda-python
 fi
 

ci/task/test_model_compile.sh

@@ -16,25 +16,25 @@ elif [[ ${GPU} == rocm* ]]; then
 	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-rocm57
 elif [[ ${GPU} == metal ]]; then
 	TARGET=metal
-	pip install --pre -U --force-reinstal -f https://mlc.ai/wheels mlc-ai-nightly
+	pip install --pre -U --force-reinstal -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 elif [[ ${GPU} == wasm* ]]; then
 	TARGET=wasm
-	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 	export TVM_SOURCE_DIR=$(dirname $(python -c 'import tvm; print(tvm.__file__)'))
 	export TVM_HOME=${TVM_SOURCE_DIR}
 	export MLC_LLM_SOURCE_DIR=$(pwd)
 	cd $TVM_SOURCE_DIR/web/ && make -j${NUM_THREADS} && cd -
 	cd $MLC_LLM_SOURCE_DIR/web/ && make -j${NUM_THREADS} && cd -
 elif [[ ${GPU} == ios ]]; then
 	TARGET=ios
-	pip install --pre -U --force-reinstal -f https://mlc.ai/wheels mlc-ai-nightly
+	pip install --pre -U --force-reinstal -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 elif [[ ${GPU} == android* ]]; then
 	TARGET=android
-	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 	source /android_env_vars.sh
 else
 	TARGET=vulkan
-	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+	pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 fi
 
 python tests/python/integration/test_model_compile.py $TARGET $NUM_THREADS

docs/install/mlc_llm.rst

@@ -30,7 +30,7 @@ Select your operating system/compute platform and run the command in your termin
                 .. code-block:: bash
 
                     conda activate your-environment
-                    python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly mlc-ai-nightly
+                    python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cpu mlc-ai-nightly-cpu
 
             .. tab:: CUDA 12.1
 
@@ -98,7 +98,7 @@ Select your operating system/compute platform and run the command in your termin
                 .. code-block:: bash
 
                     conda activate your-environment
-                    python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly mlc-ai-nightly
+                    python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cpu mlc-ai-nightly-cpu
 
         .. note::
 
@@ -124,7 +124,7 @@ Select your operating system/compute platform and run the command in your termin
                 .. code-block:: bash
 
                     conda activate your-environment
-                    python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly mlc-ai-nightly
+                    python -m pip install --pre -U -f https://mlc.ai/wheels mlc-llm-nightly-cpu mlc-ai-nightly-cpu
 
         .. note::
             Please make sure your conda environment comes with python and pip.

docs/install/tvm.rst

@@ -37,7 +37,7 @@ A nightly prebuilt Python package of Apache TVM Unity is provided.
             .. code-block:: bash
 
               conda activate your-environment
-              python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+              python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 
          .. tab:: CUDA 12.1
 
@@ -88,7 +88,7 @@ A nightly prebuilt Python package of Apache TVM Unity is provided.
             .. code-block:: bash
 
               conda activate your-environment
-              python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+              python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 
         .. note::
 
@@ -109,7 +109,7 @@ A nightly prebuilt Python package of Apache TVM Unity is provided.
             .. code-block:: bash
 
               conda activate your-environment
-              python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly
+              python -m pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
 
       .. note::
         Make sure you also install vulkan loader and clang to avoid vulkan