format code

15a93bec · fsx950223 · c127c062 · 15a93bec
Commit 15a93bec authored Jan 28, 2023 by fsx950223
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

include/ck/tensor_operation/gpu/grid/gridwise_sparse_embeddings_forward_layernorm.hpp ...gpu/grid/gridwise_sparse_embeddings_forward_layernorm.hpp +1 -3

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_sparse_embeddings_forward_layernorm.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_sparse_embeddings_forward_layernorm.hpp
@@ -185,9 +185,7 @@ struct GridwiseSparseEmbeddingsForwardLayernorm
                        },
                        Number<NumEmbeddings>{});
                    auto out_data_refs = generate_tie(
-                        [&](auto) -> auto& {
+                        [&](auto) -> auto& { return acc_thread_buf(Number<register_offset>{}); },
-                            return acc_thread_buf(Number<register_offset>{});
-                        },
                        Number<1>{});
                    unpack2(emb_elementwise_op, out_data_refs, in_data_refs);
                });