Refactor the codegen directory, pt 16 (pytorch#2492)

Summary: Pull Request resolved: pytorch#2492 - Migrate `embedding_bounds_check_*` over to `codegen/utils` Reviewed By: spcyppt Differential Revision: D55950626 fbshipit-source-id: bd1931fd7af69c68651e76eac2f75dd231756951
q10 · Apr 11, 2024 · 818a26f · 818a26f
1 parent 704159f
commit 818a26f
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
@@ -564,7 +564,7 @@ set(fbgemm_gpu_sources_static_cpu
     codegen/training/forward/embedding_forward_split_cpu.cpp
     codegen/inference/embedding_forward_quantized_host_cpu.cpp
     codegen/training/backward/embedding_backward_dense_host_cpu.cpp
-    codegen/embedding_bounds_check_host_cpu.cpp
+    codegen/utils/embedding_bounds_check_host_cpu.cpp
     src/merge_pooled_embedding_ops/merge_pooled_embedding_ops_cpu.cpp
     src/permute_pooled_embedding_ops/permute_pooled_embedding_function.cpp
     src/permute_pooled_embedding_ops/permute_pooled_embedding_ops_cpu.cpp
@@ -590,7 +590,7 @@ if(NOT FBGEMM_CPU_ONLY)
   list(APPEND fbgemm_gpu_sources_static_cpu
     codegen/inference/embedding_forward_quantized_host.cpp
     codegen/training/backward/embedding_backward_dense_host.cpp
-    codegen/embedding_bounds_check_host.cpp
+    codegen/utils/embedding_bounds_check_host.cpp
     src/memory_utils/memory_utils.cpp
     src/memory_utils/memory_utils_ops.cpp
     src/memory_utils/memory_utils_ops_cpu.cpp
@@ -628,7 +628,7 @@ endif()
 
 if(NOT FBGEMM_CPU_ONLY)
   set(fbgemm_gpu_sources_static_gpu
-      codegen/embedding_bounds_check.cu
+      codegen/utils/embedding_bounds_check.cu
       codegen/inference/embedding_forward_quantized_split_lookup.cu
       src/memory_utils/memory_utils.cu
       src/memory_utils/memory_utils_ops.cu

diff --git a/...gpu/codegen/embedding_ops_placeholder.cpp → ...en/training/embedding_ops_placeholder.cpp b/...gpu/codegen/embedding_ops_placeholder.cpp → ...en/training/embedding_ops_placeholder.cpp
diff --git a/fbgemm_gpu/codegen/embedding_bounds_check.cu → ...u/codegen/utils/embedding_bounds_check.cu b/fbgemm_gpu/codegen/embedding_bounds_check.cu → ...u/codegen/utils/embedding_bounds_check.cu
diff --git a/...u/codegen/embedding_bounds_check_host.cpp → ...gen/utils/embedding_bounds_check_host.cpp b/...u/codegen/embedding_bounds_check_host.cpp → ...gen/utils/embedding_bounds_check_host.cpp
diff --git a/...degen/embedding_bounds_check_host_cpu.cpp → ...utils/embedding_bounds_check_host_cpu.cpp b/...degen/embedding_bounds_check_host_cpu.cpp → ...utils/embedding_bounds_check_host_cpu.cpp
diff --git a/fbgemm_gpu/test/sparse/utils_test.cpp b/fbgemm_gpu/test/sparse/utils_test.cpp
@@ -12,7 +12,7 @@
 #include <ATen/ATen.h>
 #include <torch/torch.h>
 
-#include "deeplearning/fbgemm/fbgemm_gpu/include/fbgemm_gpu/sparse_ops_utils.h"
+#include "fbgemm_gpu/sparse_ops_utils.h"
 
 using namespace testing;
 

diff --git a/fbgemm_gpu/test/tbe/utils/cpu_kernel_test.cpp b/fbgemm_gpu/test/tbe/utils/cpu_kernel_test.cpp
@@ -11,8 +11,8 @@
 #include <ATen/ATen.h>
 #include <ATen/AccumulateType.h>
 
-#include "deeplearning/fbgemm/fbgemm_gpu/include/fbgemm_gpu/embedding_forward_split_cpu.h"
 #include "fbgemm_gpu/embedding_common.h"
+#include "fbgemm_gpu/embedding_forward_split_cpu.h"
 #include "torch/types.h" // @manual=//caffe2:torch-cpp-cpu
 
 TEST(CpuKernelTest, csr2csc_test) {