diff --git a/include/cute/util/print_tensor.hpp b/include/cute/util/print_tensor.hpp index aabd0d844c..ca578e4911 100644 --- a/include/cute/util/print_tensor.hpp +++ b/include/cute/util/print_tensor.hpp @@ -33,6 +33,7 @@ #include // CUTE_HOST_DEVICE #include +#include #include namespace cute diff --git a/test/self_contained_includes/CMakeLists.txt b/test/self_contained_includes/CMakeLists.txt index 97d1ce0c8f..e4563f1ed1 100644 --- a/test/self_contained_includes/CMakeLists.txt +++ b/test/self_contained_includes/CMakeLists.txt @@ -53,6 +53,7 @@ set(header_files_to_check cute/tensor.hpp cute/tensor_impl.hpp cute/underscore.hpp + cute/util/print_tensor.hpp # cute/algorithm cute/algorithm/axpby.hpp cute/algorithm/clear.hpp @@ -275,4 +276,3 @@ endforeach() # build all generated .cu files into a single library cutlass_add_library(test_self_contained_includes MODULE ${_gen_source_files}) -