diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp index bc0dc592f0..521968a43b 100644 --- a/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp +++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_default_policy.hpp @@ -415,12 +415,12 @@ struct BlockFmhaBwdPipelineDefaultPolicy constexpr index_t N2_m = kNPerBlock / (N1_m * N0); constexpr auto dstr_m = make_static_tile_distribution( tile_distribution_encoding< - sequence<>, - tuple, sequence>, - tuple, sequence<1, 2>>, // N0, N1 K1 - tuple, sequence<1, 1>>, - sequence<2, 1, 2>, // K0 N2 K2 - sequence<0, 2, 2>>{}); + sequence<>, + tuple, sequence>, + tuple, sequence<1, 2>>, // N0, N1 K1 + tuple, sequence<1, 1>>, + sequence<2, 1, 2>, // K0 N2 K2 + sequence<0, 2, 2>>{}); static_assert(container_reduce(dstr_m.get_lengths(), std::multiplies{}, 1) == kNPerBlock * kKPerBlock); return dstr_m; @@ -464,12 +464,12 @@ struct BlockFmhaBwdPipelineDefaultPolicy constexpr index_t N0_m = kNPerBlock / (N2_m * N1); constexpr auto dstr_m = make_static_tile_distribution( tile_distribution_encoding< - sequence<>, - tuple, sequence>, - tuple, sequence<1, 2>>, // N1, N2 K1 - tuple, sequence<2, 1>>, - sequence<2, 1, 2>, // K0 N0 K2 - sequence<0, 0, 2>>{}); + sequence<>, + tuple, sequence>, + tuple, sequence<1, 2>>, // N1, N2 K1 + tuple, sequence<2, 1>>, + sequence<2, 1, 2>, // K0 N0 K2 + sequence<0, 0, 2>>{}); static_assert(container_reduce(dstr_m.get_lengths(), std::multiplies{}, 1) == kNPerBlock * kKPerBlock); return dstr_m; @@ -515,12 +515,12 @@ struct BlockFmhaBwdPipelineDefaultPolicy constexpr index_t M2_m = kMPerBlock / (M1_m * M0); constexpr auto dstr_m = make_static_tile_distribution( tile_distribution_encoding< - sequence<>, - tuple, sequence>, - tuple, sequence<1, 2>>, // M0, M1 K1 - tuple, sequence<1, 1>>, - sequence<2, 1, 2>, // K0 M2 K2 - sequence<0, 2, 2>>{}); + sequence<>, + tuple, sequence>, + tuple, sequence<1, 2>>, // M0, M1 K1 + tuple, sequence<1, 1>>, + sequence<2, 1, 2>, // K0 M2 K2 + sequence<0, 2, 2>>{}); static_assert(container_reduce(dstr_m.get_lengths(), std::multiplies{}, 1) == kMPerBlock * kKPerBlock); return dstr_m; @@ -566,12 +566,12 @@ struct BlockFmhaBwdPipelineDefaultPolicy constexpr index_t M2_m = kMPerBlock / (M1_m * M0); constexpr auto dstr_m = make_static_tile_distribution( tile_distribution_encoding< - sequence<>, - tuple, sequence>, - tuple, sequence<1, 2>>, // M0, M1 K1 - tuple, sequence<1, 1>>, - sequence<2, 1, 2>, // K0 M2 K2 - sequence<0, 2, 2>>{}); + sequence<>, + tuple, sequence>, + tuple, sequence<1, 2>>, // M0, M1 K1 + tuple, sequence<1, 1>>, + sequence<2, 1, 2>, // K0 M2 K2 + sequence<0, 2, 2>>{}); static_assert(container_reduce(dstr_m.get_lengths(), std::multiplies{}, 1) == kMPerBlock * kKPerBlock); return dstr_m; diff --git a/include/ck_tile/remod.py b/include/ck_tile/remod.py index 6f5a425207..e5e5ad4300 100644 --- a/include/ck_tile/remod.py +++ b/include/ck_tile/remod.py @@ -1,8 +1,14 @@ -from datetime import datetime import pathlib from pathlib import Path import subprocess import os - import copy +from datetime import datetime +import pathlib +from pathlib import Path +import subprocess +import os +import copy - NS = 'ck_tile' OPS = 'ops' REF = 'ref' OPS_COMMON = - 'common' #common header will be duplicated into ops/* other module +NS = 'ck_tile' +OPS = 'ops' +REF = 'ref' +OPS_COMMON = 'common' #common header will be duplicated into ops/* other module HEADER_COMMON = f"""// SPDX-License-Identifier: MIT // Copyright (c) 2018-{datetime.now().year}, Advanced Micro Devices, Inc. All rights reserved.\n diff --git a/script/clang-format-overwrite.sh b/script/clang-format-overwrite.sh index 53de05a7d8..a770970fef 100755 --- a/script/clang-format-overwrite.sh +++ b/script/clang-format-overwrite.sh @@ -1,2 +1,2 @@ find . -name deps -prune -o -name build -prune -o -iname '*.h' -o -iname '*.hpp' -o -iname '*.cpp' -o -iname '*.h.in' -o -iname '*.hpp.in' -o -iname '*.cpp.in' -o -iname '*.cl' -o -iname '*.cuh' -o -iname '*.cu' -o -iname '*.inc' | xargs -n 1 -P 16 -I{} -t sh -c 'clang-format-18 -i -style=file {}' -git status --porcelain | awk '$1 != "D" && (match($2, "\\.cpp|hpp|inc")) {print $2}' | xargs -n 1 -P 16 -I{} -t sh -c 'clang-format-18 -i -style=file {}' +git status --porcelain | awk '$1 != "D" && (match($2, "\\.cpp|.hpp|.inc")) {print $2}' | xargs -n 1 -P 16 -I{} -t sh -c 'clang-format-18 -i -style=file {}'