Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "8eaaa546d89f836b716e92348786d878f883ee86"
Commit 8daff431 authored by wangshaojie6's avatar wangshaojie6
Browse files

use another instance to check the efficiency

parent 7acbf104
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 11 deletions
example/42_groupnorm/groupnorm_sigmoid_fp16.cpp
View file @ 8daff431
...@@ -55,26 +55,26 @@ using DeviceInstance = ...@@ -55,26 +55,26 @@ using DeviceInstance =
YElementOp, YElementOp,
Rank, Rank,
NumReduceDim, NumReduceDim,
256, // BlockSize 1024, // BlockSize
8, // ClusterM 1, // ClusterM
32, // ClusterK 1024, // ClusterK
1, // SliceM 1, // SliceM
8, // SliceK 8, // SliceK
1, // SrcVecDim (0=M, 1=K) 1, // SrcVecDim (0=M, 1=K)
8, // SrcScalarPerVector 2, // SrcScalarPerVector
1, // GammaVecDim (0=M, 1=K) 1, // GammaVecDim (0=M, 1=K)
8, // GammaScalarPerVector 2, // GammaScalarPerVector
1, // BetaVecDim (0=M, 1=K) 1, // BetaVecDim (0=M, 1=K)
8, // BetaScalarPerVector 2, // BetaScalarPerVector
8>; // OutScalarPerVector 2>; // OutScalarPerVector
int main(int argc, char* argv[]) int main(int argc, char* argv[])
{ {
ck::index_t N = 128; ck::index_t N = 2;
ck::index_t H = 16; ck::index_t H = 32;
ck::index_t W = 16; ck::index_t W = 32;
ck::index_t G = 32; ck::index_t G = 32;
ck::index_t C = 40; ck::index_t C = 30;
if(argc == 1) if(argc == 1)
{ {
......
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