fix build wip

example/ck_tile/01_fmha/CMakeLists.txt

@@ -14,7 +14,7 @@ add_custom_command(
   --output_dir ${CMAKE_CURRENT_BINARY_DIR}
 )
 
-set(EXAMPLE_FMHA_FWD "example_fmha_fwd")
+set(EXAMPLE_FMHA_FWD "ck_tile_example_fmha_fwd")
 add_example_executable(${EXAMPLE_FMHA_FWD} fmha_fwd.cpp)
 target_include_directories(${EXAMPLE_FMHA_FWD} PRIVATE ${CMAKE_CURRENT_LIST_DIR})
 target_sources(${EXAMPLE_FMHA_FWD} PRIVATE ${FMHA_FWD_GEN_BLOBS})

example/ck_tile/01_fmha/generate.py

@@ -5,7 +5,7 @@
 import argparse
 import itertools
 from pathlib import Path
-from typing import List, Optional, tuple
+from typing import List, Optional
 from dataclasses import dataclass
 import copy
 

example/ck_tile/01_fmha/mask.hpp

@@ -7,7 +7,7 @@
 #include <string>
 
 #include "ck_tile/core.hpp"
-#include "ck_tile/fmha.hpp"
+#include "ck_tile/ops/fmha.hpp"
 
 enum class mask_enum
 {

example/ck_tile/CMakeLists.txt

@@ -0,0 +1,5 @@
+include_directories(AFTER
+  ${CMAKE_CURRENT_LIST_DIR}
+)
+
+add_subdirectory(01_fmha)