enable support for the native tf32 dpas

Author: bashbaug

samples/99_matrixexperimentstf32/main.cpp

@@ -517,17 +517,14 @@ int main(int argc, char** argv)
     auto minSubGroupSize = findMinSubGroupSize(device);
 
     bool emulate_tN16 = true;
-    if (!emulate && checkDeviceForExtension(device, "cl_intel_subgroup_matrix_multiply_accumulate")) {
-        printf("Found support for cl_intel_subgroup_matrix_multiply_accumulate, min sub-group size is: %zu\n", minSubGroupSize);
+    if (!emulate && checkDeviceForExtension(device, "cl_intel_subgroup_matrix_multiply_accumulate_tf32")) {
+        printf("Found support for cl_intel_subgroup_matrix_multiply_accumulate_tf32, min sub-group size is: %zu\n", minSubGroupSize);
         switch(minSubGroupSize) {
             case 16: emulate_tN16 = false; break;
             default: break;
         }
     }
 
-    printf("NOTE: dpas is unconditionally emulated, currently!\n");
-    emulate_tN16 = true;
-
     buildOptions += " -DEMULATE_tN16=" + std::to_string(emulate_tN16);
 
     printf("Config:\n");

samples/99_matrixexperimentstf32/matrix_helpers_tf32.cl

@@ -71,7 +71,6 @@ float emu_sub_group_tf32_tf32_matrix_mad_k8(float  a, float8 b, float  acc)
 {
     float res = acc;
 
-#if 1
     res = fma(sub_group_broadcast(a, 0), b.s0, res);
     res = fma(sub_group_broadcast(a, 1), b.s1, res);
     res = fma(sub_group_broadcast(a, 2), b.s2, res);
@@ -80,12 +79,6 @@ float emu_sub_group_tf32_tf32_matrix_mad_k8(float  a, float8 b, float  acc)
     res = fma(sub_group_broadcast(a, 5), b.s5, res);
     res = fma(sub_group_broadcast(a, 6), b.s6, res);
     res = fma(sub_group_broadcast(a, 7), b.s7, res);
-#else
-float  __attribute__((overloadable)) intel_sub_group_tf32_tf32_matrix_mad_k8_f32(short  a, int8 b, float  acc);
-    uint    a_ui = as_uint(sub_group_shuffle(a, get_sub_group_local_id() / 2));
-    short   aData = get_sub_group_local_id() % 2 ? as_short2(a_ui).hi : as_short2(a_ui).lo;
-    res = intel_sub_group_tf32_tf32_matrix_mad_k8_f32(aData, as_int8(b), res);
-#endif
 
     return res;
 }