Re-organize Permute Pooled Embedding Ops

Summary: - Move Permute Pooled Embedding Ops code into its own directory Reviewed By: sryap Differential Revision: D47279821 fbshipit-source-id: 157f0b3519b14c4819f70b183531beca0333d2e7
pytorch · Jul 7, 2023 · 5b1f58a · 5b1f58a
1 parent b92a351
commit 5b1f58a
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 5 deletions.
diff --git a/fbgemm_gpu/CMakeLists.txt b/fbgemm_gpu/CMakeLists.txt
@@ -549,7 +549,7 @@ set(fbgemm_gpu_sources_static_cpu
     codegen/embedding_forward_quantized_host_cpu.cpp
     codegen/embedding_backward_dense_host_cpu.cpp
     codegen/embedding_bounds_check_host_cpu.cpp
-    src/permute_pooled_embedding_ops_split_cpu.cpp
+    src/permute_pooled_embedding_ops/permute_pooled_embedding_ops_split_cpu.cpp
     src/jagged_tensor_ops/jagged_tensor_ops_autograd.cpp
     src/jagged_tensor_ops/jagged_tensor_ops_meta.cpp
     src/jagged_tensor_ops/jagged_tensor_ops_cpu.cpp
@@ -567,8 +567,8 @@ if(NOT FBGEMM_CPU_ONLY)
     codegen/embedding_bounds_check_host.cpp
     src/cumem_utils_host.cpp
     src/layout_transform_ops_gpu.cpp
-    src/permute_pooled_embedding_ops_gpu.cpp
-    src/permute_pooled_embedding_ops_split_gpu.cpp
+    src/permute_pooled_embedding_ops/permute_pooled_embedding_ops_gpu.cpp
+    src/permute_pooled_embedding_ops/permute_pooled_embedding_ops_split_gpu.cpp
     src/quantize_ops/quantize_ops_gpu.cpp
     src/sparse_ops/sparse_ops_gpu.cpp
     src/split_embeddings_utils.cpp
@@ -629,8 +629,8 @@ if(NOT FBGEMM_CPU_ONLY)
       src/jagged_tensor_ops/keyed_jagged_index_select_dim1.cu
       src/layout_transform_ops.cu
       src/metric_ops.cu
-      src/permute_pooled_embedding_ops_split.cu
-      src/permute_pooled_embedding_ops.cu
+      src/permute_pooled_embedding_ops/permute_pooled_embedding_ops_split.cu
+      src/permute_pooled_embedding_ops/permute_pooled_embedding_ops.cu
       src/quantize_ops/quantize_bfloat16.cu
       src/quantize_ops/quantize_fp8_rowwise.cu
       src/quantize_ops/quantize_fused_8bit_rowwise.cu

diff --git a/...m_gpu/src/permute_pooled_embedding_ops.cu → ...dding_ops/permute_pooled_embedding_ops.cu b/...m_gpu/src/permute_pooled_embedding_ops.cu → ...dding_ops/permute_pooled_embedding_ops.cu
diff --git a/.../src/permute_pooled_embedding_ops_cpu.cpp → ..._ops/permute_pooled_embedding_ops_cpu.cpp b/.../src/permute_pooled_embedding_ops_cpu.cpp → ..._ops/permute_pooled_embedding_ops_cpu.cpp
diff --git a/.../src/permute_pooled_embedding_ops_gpu.cpp → ..._ops/permute_pooled_embedding_ops_gpu.cpp b/.../src/permute_pooled_embedding_ops_gpu.cpp → ..._ops/permute_pooled_embedding_ops_gpu.cpp
diff --git a/...src/permute_pooled_embedding_ops_split.cu → ...ops/permute_pooled_embedding_ops_split.cu b/...src/permute_pooled_embedding_ops_split.cu → ...ops/permute_pooled_embedding_ops_split.cu
diff --git a/...ermute_pooled_embedding_ops_split_cpu.cpp → ...ermute_pooled_embedding_ops_split_cpu.cpp b/...ermute_pooled_embedding_ops_split_cpu.cpp → ...ermute_pooled_embedding_ops_split_cpu.cpp
diff --git a/...ermute_pooled_embedding_ops_split_gpu.cpp → ...ermute_pooled_embedding_ops_split_gpu.cpp b/...ermute_pooled_embedding_ops_split_gpu.cpp → ...ermute_pooled_embedding_ops_split_gpu.cpp