update doc and test name

Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com>

Author: sramakintel

python/README.md

@@ -2,6 +2,8 @@
 
 [Intel® Distribution for Python*] enhances performance and can improve your program speed from 10 to 100 times faster. It is a Python* distribution that includes the [Intel® Math Kernel Library] (oneMKL) and other Intel performance libraries to enable near-native performance through acceleration of core numerical and machine learning packages.
 
+Intel® Distribution for Python* is also available for Intel® dGPUs, that includes the latest driver packages and Intel® OneAPI runtime libraries that enables Machine Learning frameworks leverage the XPU device plugin. 
+
 ## Images
 
 The images below include variations for only the core packages in the [Intel® Distribution for Python*] installation, or all of the packages.
@@ -10,6 +12,7 @@ The images below include variations for only the core packages in the [Intel® D
 | ---------------------- | ---------- |
 | `3.10-full`, `latest`  | `2024.2.0` |
 | `3.10-core`            | `2024.2.0` |
+| `3.10-xpu-idp`         | `2024.2.1` |
 
 ## Run a Performance Sample
 
@@ -38,6 +41,19 @@ pip install numpy
 python /tests/perf_sample.py
 ```
 
+### Run a Sanity test using the XPU variant
+
+Use the following command to check the availability of Intel dGPU devices on your system and the presence of Intel® OneAPI runtime libraries.
+
+```bash
+# Use the working directory from the first command
+docker run --rm -it \
+    -v $PWD/tests:/tests \
+    --device /dev/dri \
+    intel/python:3.10-xpu-idp \
+    bash /tests/xpu_base_layers_test.sh
+```
+
 ## Build from Source (Advanced)
 
 To build the images from source, clone the [AI Containers](https://github.com/intel/ai-containers) repository, follow the main `README.md` file to setup your environment, and run the following command:
@@ -48,12 +64,21 @@ docker compose build idp
 docker compose run idp
 ```
 
+To build the xpu variant of the image, run the following commands:
+
+```bash
+cd python
+docker compose build xpu
+docker compose run xpu
+```
+
 You can find the list of services below for each container in the group:
 
 | Service Name | Description                                                         |
 | ------------ | ------------------------------------------------------------------- |
 | `idp`        | Base image with [Intel® Distribution for Python*]                    |
 | `pip`        | Equivalent python image without [Intel® Distribution for Python*]    |
+| `xpu`        | Base Image for Intel XPU plugin with and without [Intel® Distribution for Python*] |
 
 ## License
 
@@ -67,3 +92,6 @@ It is the image user's responsibility to ensure that any use of The images below
 
 [Intel® Distribution for Python*]: https://www.intel.com/content/www/us/en/developer/tools/oneapi/distribution-for-python.html#gs.9bos9m
 [Intel® Math Kernel Library]: https://www.intel.com/content/www/us/en/developer/tools/oneapi/onemkl.html
+[Intel® DPC++ Compiler Library]: https://www.intel.com/content/www/us/en/developer/tools/oneapi/dpc-compiler-download.html
+[Intel® Collective Communications Library]: https://www.intel.com/content/www/us/en/developer/tools/oneapi/oneccl.html
+[Intel® dGPU driver releases ]: https://dgpu-docs.intel.com/releases/releases.html

python/tests/tests.yaml

@@ -27,7 +27,7 @@ perf-stock:
   volumes:
   - dst: /tests
     src: $PWD/python/tests
-xpu-packages-${PACKAGE_OPTION:-pip}:
+xpu-base-layers-${PACKAGE_OPTION:-pip}:
   img: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-xpu-base
   cmd: bash /tests/xpu_packages_test.sh
   device: ["/dev/dri"]