fix header for paddle

Signed-off-by: Phuong Nguyen <[email protected]>
NVIDIA · Dec 3, 2024 · dedd04b · dedd04b
1 parent 2d5d355
commit dedd04b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/transformer_engine/paddle/csrc/common.h b/transformer_engine/paddle/csrc/common.h
@@ -10,9 +10,8 @@
 #include <transformer_engine/cast.h>
 #include <transformer_engine/fused_attn.h>
 #include <transformer_engine/gemm.h>
-#include <transformer_engine/layer_norm.h>
+#include <transformer_engine/normalization.h>
 #include <transformer_engine/recipe.h>
-#include <transformer_engine/rmsnorm.h>
 #include <transformer_engine/softmax.h>
 #include <transformer_engine/transformer_engine.h>
 #include <transformer_engine/transpose.h>