Split Sparse Ops, pt. 1 (#1843)

Summary: Pull Request resolved: #1843 - Migrate sparse ops code into its own directory Reviewed By: sryap Differential Revision: D46922993 fbshipit-source-id: b494faaea7883c0f0638771c5f8ef871d7b7448e
pytorch · Jun 25, 2023 · 8979de3 · 8979de3
1 parent 751ce3a
commit 8979de3
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 4 deletions.
diff --git a/fbgemm_gpu/CMakeLists.txt b/fbgemm_gpu/CMakeLists.txt
@@ -556,8 +556,8 @@ set(fbgemm_gpu_sources_static_cpu
     src/input_combine_cpu.cpp
     src/layout_transform_ops_cpu.cpp
     src/quantize_ops_cpu.cpp
-    src/sparse_ops_cpu.cpp
-    src/sparse_ops_meta.cpp
+    src/sparse_ops/sparse_ops_cpu.cpp
+    src/sparse_ops/sparse_ops_meta.cpp
     src/embedding_inplace_update_cpu.cpp)
 
 if(NOT FBGEMM_CPU_ONLY)
@@ -570,7 +570,7 @@ if(NOT FBGEMM_CPU_ONLY)
     src/permute_pooled_embedding_ops_gpu.cpp
     src/permute_pooled_embedding_ops_split_gpu.cpp
     src/quantize_ops_gpu.cpp
-    src/sparse_ops_gpu.cpp
+    src/sparse_ops/sparse_ops_gpu.cpp
     src/split_embeddings_utils.cpp
     src/split_table_batched_embeddings.cpp
     src/metric_ops_host.cpp
@@ -635,7 +635,7 @@ if(NOT FBGEMM_CPU_ONLY)
       src/permute_pooled_embedding_ops_split.cu
       src/permute_pooled_embedding_ops.cu
       src/quantize_ops.cu
-      src/sparse_ops.cu
+      src/sparse_ops/sparse_ops.cu
       src/split_embeddings_cache_cuda.cu
       src/split_embeddings_utils.cu)
 

diff --git a/fbgemm_gpu/src/sparse_ops.cu → fbgemm_gpu/src/sparse_ops/sparse_ops.cu b/fbgemm_gpu/src/sparse_ops.cu → fbgemm_gpu/src/sparse_ops/sparse_ops.cu
diff --git a/fbgemm_gpu/src/sparse_ops_cpu.cpp → fbgemm_gpu/src/sparse_ops/sparse_ops_cpu.cpp b/fbgemm_gpu/src/sparse_ops_cpu.cpp → fbgemm_gpu/src/sparse_ops/sparse_ops_cpu.cpp
diff --git a/fbgemm_gpu/src/sparse_ops_gpu.cpp → fbgemm_gpu/src/sparse_ops/sparse_ops_gpu.cpp b/fbgemm_gpu/src/sparse_ops_gpu.cpp → fbgemm_gpu/src/sparse_ops/sparse_ops_gpu.cpp
diff --git a/fbgemm_gpu/src/sparse_ops_meta.cpp → ...mm_gpu/src/sparse_ops/sparse_ops_meta.cpp b/fbgemm_gpu/src/sparse_ops_meta.cpp → ...mm_gpu/src/sparse_ops/sparse_ops_meta.cpp