try fixing CI

codeplaysoftware · Feb 21, 2025 · 7fe0080 · 7fe0080
1 parent 25b486f
commit 7fe0080
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tools/copy_debug/copy_debug.cpp b/tools/copy_debug/copy_debug.cpp
@@ -29,7 +29,8 @@
  *
  **************************************************************************************************/
 
-#include "cutlass/detail/layout.hpp"
+//#include "cutlass/detail/layout.hpp"
+//#include "cute/atom/copy_traits_xe.hpp"
 
 #include <cute/tensor.hpp>
 #include <sycl/sycl.hpp>
@@ -40,6 +41,7 @@
 #include <cutlass/cutlass.h>
 
 using namespace syclcompat::experimental;
+using namespace cute;
 
 #define SUBGROUP_SIZE (16)
 
@@ -63,7 +65,7 @@ void copy_kernel(TensorS S) {
     Layout<CopyThreadShape>{},
     make_layout(shape_div(typename traits_load::BlockShape{}, CopyThreadShape{})));
 
-  auto thr_copy_load = tiled_copy_load.get_slice(cutlass::get_sub_group_local_id());
+  auto thr_copy_load = tiled_copy_load.get_slice(ThreadIdxX());
 
   using actual_fragment_size = std::conditional_t<std::is_same_v<fragment_size, void>, C<Atom_load::NumValDst>, fragment_size>;
   Tensor fragment = make_tensor<Element>(make_shape(actual_fragment_size{},_1{},_1{}));