[Matrix] Add new feature flag 'matrix-fp16' (#11667)

If the target device is CPU, the flag requires the support of AMX-FP16
ISA (Granite Rapids or later).

'matrix-xmx8' and 'gpu-intel-pvc' both should imply 'matrix-fp16'.

'gpu-intel-pvc' also implies 'matrix-tf32'.

Signed-off-by: Yilong Guo <[email protected]>

---------

Signed-off-by: Yilong Guo <[email protected]>

Author: Nuullll

sycl/test-e2e/Matrix/Legacy/XMX8/element_wise_all_ops_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix-xmx8
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=1
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/Legacy/XMX8/joint_matrix_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix-xmx8
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=1
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/Legacy/element_wise_all_ops_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=1
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/Legacy/joint_matrix_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=1
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/SG32/element_wise_all_ops_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix,gpu
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/SG32/joint_matrix_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/XMX8/element_wise_all_ops_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix-xmx8
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/XMX8/joint_matrix_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix-xmx8
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/element_wise_all_ops_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
 // RUN: %{run} %t.out

sycl/test-e2e/Matrix/joint_matrix_half.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 // REQUIRES: aspect-fp16
 // REQUIRES: matrix
+// REQUIRES: matrix-fp16
 
 // RUN: %{build} -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
 // RUN: %{run} %t.out

sycl/test-e2e/lit.cfg.py

@@ -119,6 +119,8 @@
     config.available_features.add('gpu-intel-dg2')
 if lit_config.params.get('gpu-intel-pvc', False):
     config.available_features.add('gpu-intel-pvc')
+    config.available_features.add('matrix-fp16') # PVC implies the support of FP16 matrix
+    config.available_features.add('matrix-tf32') # PVC implies the support of TF32 matrix
 
 if lit_config.params.get('matrix', False):
     config.available_features.add('matrix')
@@ -128,6 +130,10 @@
 
 if lit_config.params.get('matrix-xmx8', False):
     config.available_features.add('matrix-xmx8')
+    config.available_features.add('matrix-fp16') # XMX implies the support of FP16 matrix
+
+if lit_config.params.get('matrix-fp16', False):
+    config.available_features.add('matrix-fp16')
 
 #support for LIT parameter ze_debug<num>
 if lit_config.params.get('ze_debug'):