save code

Author: xiaolil1

csrc/xpu_cutlass_fusion.cpp

@@ -417,24 +417,24 @@ printf("src_compress_size = %d, dst_compress_size = %d, src_vec_size = %d, dst_v
         constexpr int src_loop_num = 1; //K / src_vec_size / src_compress_size;
         constexpr int dst_loop_num = 1; //K / dst_vec_size / dst_compress_size;
 
-        src_compress_type src[src_loop_num * src_vec_size];
+        //src_compress_type src[src_loop_num * src_vec_size];
         ElementMMA dst[dst_loop_num * dst_compress_size * dst_vec_size];
 
-        reinterpret_cast<sycl::vec<src_compress_type, src_vec_size>*>(src)[0] = reinterpret_cast<sycl::vec<src_compress_type, src_vec_size>*>(cute::raw_pointer_cast(dequant_frag.data()))[0];
+        //reinterpret_cast<sycl::vec<src_compress_type, src_vec_size>*>(src)[0] = reinterpret_cast<sycl::vec<src_compress_type, src_vec_size>*>(cute::raw_pointer_cast(dequant_frag.data()))[0];
         float scale_value = fragment_scale(0);//(dst_base_idx + c) >> (31 - std::countl_zero<unsigned int>(GROUP_SIZE)));
 
         #pragma unroll
         for (int v = 0; v < src_vec_size; v++) {
           int dst_base_idx = v * src_compress_size;
           int c = 0;
-          uint8_t bit_value = (src[v] >> (4 * (((c + 1) & 1) + (c >> 1) * 2))) & 0xF;
+          uint8_t bit_value = (reinterpret_cast<src_compress_type*>(cute::raw_pointer_cast(dequant_frag.data()))[v] >> (4 * (((c + 1) & 1) + (c >> 1) * 2))) & 0xF;
           float converted_value_1 = quant_map[bit_value];
           float converted_value_2 = 0.f;
           #pragma unroll
           for (; c < src_compress_size-1;) {
               converted_value_2 = converted_value_1;
               c++;
-              bit_value = (src[v] >> (4 * (((c + 1) & 1) + (c >> 1) * 2))) & 0xF;
+              bit_value = (reinterpret_cast<src_compress_type*>(cute::raw_pointer_cast(dequant_frag.data()))[v] >> (4 * (((c + 1) & 1) + (c >> 1) * 2))) & 0xF;
               converted_value_1 = quant_map[bit_value];
               dst[dst_base_idx + c-1] = static_cast<ElementMMA>(converted_value_2 * scale_value);
           }