Skip to content

Commit 90bf1f3

Browse files
committed
Fixup test_unit_cute_intel_xe
1 parent 0e231c3 commit 90bf1f3

File tree

3 files changed

+6
-11
lines changed

3 files changed

+6
-11
lines changed

test/unit/CMakeLists.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,8 +95,7 @@ function(cutlass_test_unit_add_executable NAME)
9595
PUBLIC
9696
GTest::gtest
9797
# TODO: This change works for resolving 'cutlasscompat.hpp' not found issue, fix this if it blocks merging
98-
# cutlass_test_unit_infra
99-
# cutlass_test_unit_infra_lib
98+
cutlass_tools_util_includes
10099
)
101100
else()
102101
target_link_libraries(

test/unit/cute/core/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929
cutlass_test_unit_add_executable(
3030
cutlass_test_unit_cute_core
31-
# WITHOUT_CUDA
31+
WITHOUT_CUDA
3232
array_subbyte.cpp
3333
bitfield.cpp
3434
coalesce.cpp

test/unit/cute/intel_xe/copy_block.cpp

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -141,8 +141,7 @@ struct copy_op<dtype, load, store, M, N, false> {
141141
//
142142
launch<
143143
copy_kernel_vectorized<decltype(S), decltype(D), decltype(tiled_load),
144-
decltype(tiled_store), load>, CopyKernelVectorizedName<decltype(S), decltype(D), decltype(tiled_load),
145-
decltype(tiled_store), load>>(
144+
decltype(tiled_store), load>, CopyKernelVectorizedName<load, Int<M>>>(
146145
launch_policy{
147146
cutlasscompat::dim3(1), blockDim,
148147
kernel_properties{sycl_exp::sub_group_size<SUBGROUP_SIZE>}},
@@ -194,8 +193,7 @@ struct copy_op<char, load, XE_2D_U8x2x32_ST_N, M, N, false> {
194193
//
195194
launch<
196195
copy_kernel_vectorized<decltype(S), decltype(D), decltype(tiled_load),
197-
decltype(tiled_store), load>, CopyKernelVectorizedName<decltype(S), decltype(D), decltype(tiled_load),
198-
decltype(tiled_store), load>>(
196+
decltype(tiled_store), load>, CopyKernelVectorizedName<load>>(
199197
launch_policy{
200198
cutlasscompat::dim3(1), blockDim,
201199
kernel_properties{sycl_exp::sub_group_size<SUBGROUP_SIZE>}},
@@ -247,8 +245,7 @@ struct copy_op<uint16_t, load, XE_2D_U16x2x16_ST_N, M, N, false> {
247245
//
248246
launch<
249247
copy_kernel_vectorized<decltype(S), decltype(D), decltype(tiled_load),
250-
decltype(tiled_store), load>, CopyKernelVectorizedName<decltype(S), decltype(D), decltype(tiled_load),
251-
decltype(tiled_store), load>>(
248+
decltype(tiled_store), load>, CopyKernelVectorizedName<load>>(
252249
launch_policy{
253250
cutlasscompat::dim3(1), blockDim,
254251
kernel_properties{sycl_exp::sub_group_size<SUBGROUP_SIZE>}},
@@ -310,8 +307,7 @@ struct copy_op<uint32_t, load, store, M_, N_, true> {
310307
//
311308
launch<
312309
copy_kernel_vectorized<decltype(S), decltype(D), decltype(tiled_load),
313-
decltype(tiled_store), load>, CopyKernelVectorizedName<decltype(S), decltype(D), decltype(tiled_load),
314-
decltype(tiled_store), load>>(
310+
decltype(tiled_store), load>, CopyKernelVectorizedName<load, Int<M>>>(
315311
launch_policy{
316312
cutlasscompat::dim3(1), blockDim,
317313
kernel_properties{sycl_exp::sub_group_size<SUBGROUP_SIZE>}},

0 commit comments

Comments
 (0)