ZEBinary: Add implicit argument type "work_dimensions"

ZEBinary: support work_dimensions implicit argument and execution_env
attribute has_dpas

Author: DianaChen

IGC/Compiler/CISACodeGen/OpenCLKernelCodeGen.cpp

@@ -533,6 +533,12 @@ namespace IGC
                  payloadPosition, iOpenCL::DATA_PARAMETER_DATA_SIZE * 3);
              break;
 
+         case KernelArg::ArgType::IMPLICIT_WORK_DIM:
+             zebin::ZEInfoBuilder::addPayloadArgumentImplicit(m_kernelInfo.m_zePayloadArgs,
+                 zebin::PreDefinedAttrGetter::ArgType::work_dimensions,
+                 payloadPosition, iOpenCL::DATA_PARAMETER_DATA_SIZE);
+             break;
+
         // pointer args
         case KernelArg::ArgType::PTR_GLOBAL:
         case KernelArg::ArgType::PTR_CONSTANT: {

IGC/ZEBinWriter/zebin/source/autogen/ZEInfo.hpp

@@ -36,6 +36,7 @@ struct zeInfoExecutionEnv
     zeinfo_int32_t grf_count = 0;
     zeinfo_bool_t has_4gb_buffers = false;
     zeinfo_bool_t has_device_enqueue = false;
+    zeinfo_bool_t has_dpas = false;
     zeinfo_bool_t has_fence_for_image_access = false;
     zeinfo_bool_t has_global_atomics = false;
     zeinfo_bool_t has_multi_scratch_spaces = false;
@@ -107,13 +108,14 @@ struct zeInfoContainer
     KernelsTy kernels;
 };
 struct PreDefinedAttrGetter{
-    static zeinfo_str_t getVersionNumber() { return "1.4"; }
+    static zeinfo_str_t getVersionNumber() { return "1.5"; }
 
     enum class ArgType {
         packed_local_ids,
         local_id,
         local_size,
         group_count,
+        work_dimensions,
         global_size,
         enqueued_local_size,
         global_id_offset,
@@ -161,6 +163,8 @@ struct PreDefinedAttrGetter{
             return "local_size";
         case ArgType::group_count:
             return "group_count";
+        case ArgType::work_dimensions:
+            return "work_dimensions";
         case ArgType::global_size:
             return "global_size";
         case ArgType::enqueued_local_size:

IGC/ZEBinWriter/zebin/source/autogen/ZEInfoYAML.cpp

@@ -42,6 +42,7 @@ void MappingTraits<zeInfoExecutionEnv>::mapping(IO& io, zeInfoExecutionEnv& info
     io.mapRequired("grf_count", info.grf_count);
     io.mapOptional("has_4gb_buffers", info.has_4gb_buffers, false);
     io.mapOptional("has_device_enqueue", info.has_device_enqueue, false);
+    io.mapOptional("has_dpas", info.has_dpas, false);
     io.mapOptional("has_fence_for_image_access", info.has_fence_for_image_access, false);
     io.mapOptional("has_global_atomics", info.has_global_atomics, false);
     io.mapOptional("has_multi_scratch_spaces", info.has_multi_scratch_spaces, false);

IGC/ZEBinWriter/zebin/spec/zeinfo.md

@@ -7,7 +7,7 @@ SPDX-License-Identifier: MIT
 ============================= end_copyright_notice ==========================-->
 
 # ZE Info
-Version 1.4
+Version 1.5
 
 ## Grammar
 
@@ -138,6 +138,7 @@ Supported <argument_type> of payload_arguments or per_thread_payload_arguments.
 | local_id | int16 x N x n | N is the simd_size <br> n is the number of dimensions derived from work_group_walk_order_dimensions <br> Per id have to be GRF aligned |
 | local_size | int32x3 | Number of work-items in a group |
 | group_count | int32x3 | Number of group |
+| work_dimensions | int32 | Work dimensions |
 | global_size | int32x3 | OpenCL specific feacture. The total number of work-items in each dimension |
 | enqueued_local_size | int32x3 | OpenCL specific feature. The size returned by OCL get_enqueued_local_size API  |
 | global_id_offset | int32x3 | |
@@ -256,6 +257,7 @@ Format: \<_Major number_\>.\<_Minor number_\>
 - Minor number: Increase when backward-compatible features are added. For example, add new attributes.
 
 ## Change Note
+- **Version 1.5**: Add payload_argument type work_dimensions.
 - **Version 1.4**: Add sampler_index to payload arguments.
 - **Version 1.3**: Add printf_buffer to argument_type.
 - **Version 1.2**: Add buffer_offset to argument_type.