Convert non-kernel cuda files to cpp (#1322)

* Fix file extensions Signed-off-by: Kirthi Shankar Sivamani <[email protected]> * fix build Signed-off-by: Kirthi Shankar Sivamani <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * upgrade paddle container for CI Signed-off-by: Kirthi Shankar Sivamani <[email protected]> --------- Signed-off-by: Kirthi Shankar Sivamani <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
NVIDIA · Nov 11, 2024 · 68adf45 · 68adf45
1 parent bfddb48
commit 68adf45
Show file tree

Hide file tree

Showing 15 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -76,7 +76,7 @@ jobs:
     name: 'PaddlePaddle'
     runs-on: ubuntu-latest
     container:
-      image: nvcr.io/nvidia/paddlepaddle:24.07-py3
+      image: nvcr.io/nvidia/paddlepaddle:24.10-py3
       options: --user root
     steps:
       - name: 'Checkout'

diff --git a/build_tools/paddle.py b/build_tools/paddle.py
@@ -25,7 +25,7 @@ def setup_paddle_extension(
     # Source files
     csrc_source_files = Path(csrc_source_files)
     sources = [
-        csrc_source_files / "extensions.cu",
+        csrc_source_files / "extensions.cpp",
         csrc_source_files / "common.cpp",
         csrc_source_files / "custom_ops.cu",
     ]

diff --git a/build_tools/pytorch.py b/build_tools/pytorch.py
@@ -26,7 +26,7 @@ def setup_pytorch_extension(
     csrc_source_files = Path(csrc_source_files)
     extensions_dir = csrc_source_files / "extensions"
     sources = [
-        csrc_source_files / "common.cu",
+        csrc_source_files / "common.cpp",
         csrc_source_files / "ts_fp8_op.cpp",
     ] + all_files_in_dir(extensions_dir)
 

diff --git a/transformer_engine/paddle/csrc/extensions.cu → ...sformer_engine/paddle/csrc/extensions.cpp b/transformer_engine/paddle/csrc/extensions.cu → ...sformer_engine/paddle/csrc/extensions.cpp
diff --git a/transformer_engine/pytorch/csrc/common.cu → transformer_engine/pytorch/csrc/common.cpp b/transformer_engine/pytorch/csrc/common.cu → transformer_engine/pytorch/csrc/common.cpp
@@ -5,6 +5,7 @@
  ************************************************************************/
 
 #include "common.h"
+
 #include "transformer_engine/transformer_engine.h"
 
 transformer_engine::DType getTransformerEngineFP8Type(bool e4m3_if_hybrid,

diff --git a/...ine/pytorch/csrc/extensions/activation.cu → ...ne/pytorch/csrc/extensions/activation.cpp b/...ine/pytorch/csrc/extensions/activation.cu → ...ne/pytorch/csrc/extensions/activation.cpp
diff --git a/...ine/pytorch/csrc/extensions/apply_rope.cu → ...ne/pytorch/csrc/extensions/apply_rope.cpp b/...ine/pytorch/csrc/extensions/apply_rope.cu → ...ne/pytorch/csrc/extensions/apply_rope.cpp
diff --git a/...er_engine/pytorch/csrc/extensions/cast.cu → ...r_engine/pytorch/csrc/extensions/cast.cpp b/...er_engine/pytorch/csrc/extensions/cast.cu → ...r_engine/pytorch/csrc/extensions/cast.cpp
diff --git a/...er_engine/pytorch/csrc/extensions/gemm.cu → ...r_engine/pytorch/csrc/extensions/gemm.cpp b/...er_engine/pytorch/csrc/extensions/gemm.cu → ...r_engine/pytorch/csrc/extensions/gemm.cpp
diff --git a/...er_engine/pytorch/csrc/extensions/misc.cu → ...r_engine/pytorch/csrc/extensions/misc.cpp b/...er_engine/pytorch/csrc/extensions/misc.cu → ...r_engine/pytorch/csrc/extensions/misc.cpp
diff --git a/.../pytorch/csrc/extensions/normalization.cu → ...pytorch/csrc/extensions/normalization.cpp b/.../pytorch/csrc/extensions/normalization.cu → ...pytorch/csrc/extensions/normalization.cpp
diff --git a/...engine/pytorch/csrc/extensions/padding.cu → ...ngine/pytorch/csrc/extensions/padding.cpp b/...engine/pytorch/csrc/extensions/padding.cu → ...ngine/pytorch/csrc/extensions/padding.cpp
diff --git a/..._engine/pytorch/csrc/extensions/recipe.cu → ...engine/pytorch/csrc/extensions/recipe.cpp b/..._engine/pytorch/csrc/extensions/recipe.cu → ...engine/pytorch/csrc/extensions/recipe.cpp
diff --git a/...engine/pytorch/csrc/extensions/softmax.cu → ...ngine/pytorch/csrc/extensions/softmax.cpp b/...engine/pytorch/csrc/extensions/softmax.cu → ...ngine/pytorch/csrc/extensions/softmax.cpp
diff --git a/...gine/pytorch/csrc/extensions/transpose.cu → ...ine/pytorch/csrc/extensions/transpose.cpp b/...gine/pytorch/csrc/extensions/transpose.cu → ...ine/pytorch/csrc/extensions/transpose.cpp