Fix NVTX name for LN backward (#55)

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com> Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

Fix NVTX name for LN backward (#55)
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com> Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
f22929cc · Kirthi Shankar Sivamani · GitHub · 6c9ce179 · f22929cc
Unverified Commit f22929cc authored Jan 19, 2023 by Kirthi Shankar Sivamani Committed by GitHub Jan 19, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

transformer_engine/common/layer_norm/ln_api.cpp transformer_engine/common/layer_norm/ln_api.cpp +1 -1

No files found.
--- a/transformer_engine/common/layer_norm/ln_api.cpp
+++ b/transformer_engine/common/layer_norm/ln_api.cpp
@@ -403,7 +403,7 @@ void nvte_layernorm_bwd(const NVTETensor dz,       // BxSxhidden_size
                        const int multiprocessorCount,
                        NVTETensor workspace,
                        NVTETensor barrier) {
-  NVTE_API_CALL(nvte_layernorm_fwd);
+  NVTE_API_CALL(nvte_layernorm_bwd);
  using namespace transformer_engine;
  layernorm_bwd(*reinterpret_cast<const Tensor*>(dz),
                *reinterpret_cast<const Tensor*>(x),