Modified is_C_load condition

Signed-off-by: Chawla, Amit K <[email protected]>

Author: amitchawla1

include/cutlass/epilogue/collective/xe_array_epilogue.hpp

@@ -435,11 +435,9 @@ class CollectiveEpilogue<
 
         //Instead of calling is_C_load_needed. We do heirachical check 
         //so that runtime check not there when ElementC is void
-        if constexpr (is_source_supported) {
-          if (fusion_callbacks.is_C_load_needed()) {
-            //cordinates for C and D are the same
-            copy(params.xe_load_c.with(get<0>(load_store_tensors)), tCgD(_, epi_m, epi_n), trC);
-          }
+        if (is_C_load_needed) {
+          //cordinates for C and D are the same
+          copy(params.xe_load_c.with(get<0>(load_store_tensors)), tCgD(_, epi_m, epi_n), trC);
         }
 
         cst_callbacks.previsit(epi_m, epi_n, 0, is_C_load_needed);

include/cutlass/epilogue/collective/xe_epilogue.hpp

@@ -413,10 +413,8 @@ class CollectiveEpilogue<
 
         //Instead of calling is_C_load_needed. We do heirachical check 
         //so that runtime check not there when ElementC is void
-        if constexpr (is_source_supported) {
-          if (fusion_callbacks.is_C_load_needed()) {
-            copy(params.xe_load_c, tCgC(_, epi_m, epi_n), trC);
-          }
+        if (is_C_load_needed) {
+          copy(params.xe_load_c, tCgC(_, epi_m, epi_n), trC);
         }
 
         cst_callbacks.previsit(epi_m, epi_n, 0, is_C_load_needed);