Split Sparse Ops, pt. 1 (pytorch#1843)

Summary: Pull Request resolved: pytorch#1843 - Migrate sparse ops code into its own directory Differential Revision: D46922993 fbshipit-source-id: c6341bad84f618c8d050d2902f48f935a31586c4
q10 · Jun 23, 2023 · 005619c · 005619c
1 parent eefee16
commit 005619c
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