Skip to content

GitLab

"torchvision/vscode:/vscode.git/clone" did not exist on "138b5c259578e2d4e2e2c4f538beac7a1652a551"
Commit 4831071c authored by Eddie Yan's avatar Eddie Yan
Browse files

Update layer_norm_cuda_kernel.cu

parent cdc614cf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
megatron/fused_kernels/layer_norm_cuda_kernel.cu
View file @ 4831071c
...@@ -329,6 +329,7 @@ void cuApplyLayerNorm( ...@@ -329,6 +329,7 @@ void cuApplyLayerNorm(
mean[i1] = mu; mean[i1] = mu;
invvar[i1] = c_invvar; invvar[i1] = c_invvar;
} }
__syncthreads();
} }
} }
......
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