Fix all framework build from PR 1666 (#1857)

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

Fix all framework build from PR 1666 (#1857)
Remove extra include Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
fab71571 · Kirthi Shankar Sivamani · GitHub · beffb297 · fab71571
Unverified Commit fab71571 authored Jun 06, 2025 by Kirthi Shankar Sivamani Committed by GitHub Jun 06, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

transformer_engine/jax/csrc/extensions/gemm.cpp transformer_engine/jax/csrc/extensions/gemm.cpp +0 -1

No files found.
--- a/transformer_engine/jax/csrc/extensions/gemm.cpp
+++ b/transformer_engine/jax/csrc/extensions/gemm.cpp
@@ -10,7 +10,6 @@
 #include "../extensions.h"
 #include "common/util/cuda_runtime.h"
 #include "common/util/system.h"
-#include "extensions.h"
 #include "transformer_engine/multi_stream.h"
 #include "xla/ffi/api/c_api.h"