"examples/research_projects/controlnetxs/controlnetxs.py" did not exist on "32ff4773d4b6662ddbb35c4a75f7178eb2b70cf0"
Commit 3b290001 authored by rocking's avatar rocking
Browse files

Refine variable name

parent d5efa5e5
......@@ -4,16 +4,16 @@
#include "layernorm2d_fwd_instance_common.hpp"
template <ck_tile::index_t NRepeat,
template <ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize,
ck_tile::index_t kkVectorAccessSize,
bool kTwoPass>
using t = layernorm2d_fwd_traits_<ck_tile::bf16_t,
NRepeat,
kNRepeat,
kMThreadPerBlock,
kNThreadPerBlock,
VectorAccessSize,
kkVectorAccessSize,
false,
false,
kTwoPass>;
......
......@@ -4,16 +4,16 @@
#include "layernorm2d_fwd_instance_common.hpp"
template <ck_tile::index_t NRepeat,
template <ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize,
ck_tile::index_t kVectorAccessSize,
bool kTwoPass>
using t = layernorm2d_fwd_traits_<ck_tile::bf16_t,
NRepeat,
kNRepeat,
kMThreadPerBlock,
kNThreadPerBlock,
VectorAccessSize,
kVectorAccessSize,
true,
false,
kTwoPass>;
......
......@@ -4,16 +4,16 @@
#include "layernorm2d_fwd_instance_common.hpp"
template <ck_tile::index_t NRepeat,
template <ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize,
ck_tile::index_t kVectorAccessSize,
bool kTwoPass>
using t = layernorm2d_fwd_traits_<ck_tile::fp16_t,
NRepeat,
kNRepeat,
kMThreadPerBlock,
kNThreadPerBlock,
VectorAccessSize,
kVectorAccessSize,
false,
false,
kTwoPass>;
......
......@@ -4,16 +4,16 @@
#include "layernorm2d_fwd_instance_common.hpp"
template <ck_tile::index_t NRepeat,
template <ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize,
ck_tile::index_t kVectorAccessSize,
bool kTwoPass>
using t = layernorm2d_fwd_traits_<ck_tile::fp16_t,
NRepeat,
kNRepeat,
kMThreadPerBlock,
kNThreadPerBlock,
VectorAccessSize,
kVectorAccessSize,
true,
false,
kTwoPass>;
......
......@@ -51,10 +51,10 @@ struct layernorm2d_fwd_args
// this is used to pattern-match internl kernel implementation, not to instantiate kernel
template <typename DataType_,
ck_tile::index_t NRepeat,
ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize,
ck_tile::index_t kVectorAccessSize,
bool kPadN_,
bool kSaveMeanInvStd_,
bool kTwoPass_>
......@@ -69,11 +69,11 @@ struct layernorm2d_fwd_traits_
kMThreadPerBlock * kNThreadPerBlock / warpSize;
// kNThreadPerBlock / 16;
using thread_tile = ck_tile::sequence<MRepeat, NRepeat, VectorAccessSize>;
using thread_tile = ck_tile::sequence<MRepeat, kNRepeat, kVectorAccessSize>;
using warp_tile = ck_tile::sequence<MRepeat * warpSize / kNThreadPerBlock,
NRepeat * kNThreadPerBlock * VectorAccessSize>;
kNRepeat * kNThreadPerBlock * kVectorAccessSize>;
using block_tile = ck_tile::sequence<kMWarpPerBlock * MRepeat * warpSize / kNThreadPerBlock,
NRepeat * kNThreadPerBlock * VectorAccessSize>;
kNRepeat * kNThreadPerBlock * kVectorAccessSize>;
using Shape = ck_tile::TileLayernorm2dShape<thread_tile, warp_tile, block_tile>;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment