Compiling fix

example/ck_tile/01_unified_attention/example_unified_attention.cpp

@@ -22,8 +22,7 @@
 #include <ck_tile/host/reference/reference_batched_masking.hpp>
 #include <ck_tile/host/reference/reference_batched_softmax.hpp>
 
-#include "fmha_fwd.hpp"
-#include "fmha_fwd_v3.hpp"
+#include "unified_attention.hpp"
 #include "mask.hpp"
 
 auto parse_cmd_args(int argc, char* argv[]) -> std::pair<bool, ck_tile::ArgParser>

example/ck_tile/01_unified_attention/unified_attention_runner.hpp

@@ -5,7 +5,7 @@
 
 #include "ck_tile/host.hpp"
 #include "ck_tile/ref/naive_attention.hpp"
-#include "fmha_fwd.hpp"
+#include "unified_attention.hpp"
 #include "utils.hpp"
 #include "ck_tile/utility/json_dump.hpp"
 

include/ck_tile/ops/unified_attention/pipeline/tile_unified_attention_traits.hpp

@@ -21,4 +21,4 @@ struct TileUnifiedAttentionTraits
     static constexpr bool kStoreLSE      = kStoreLSE_;
     static constexpr index_t kBlockPerCu = kBlockPerCu_;
 };
-}
+}

include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline_problem.hpp

@@ -57,4 +57,4 @@ struct UnifiedAttentionPipelineProblem
     static constexpr bool kDoFp8StaticQuant = Traits::kDoFp8StaticQuant;
     static constexpr index_t kBlockPerCu    = Traits::kBlockPerCu;
 };
-}
+}