Add FB python sources into genai CMakeLists.txt (#3886)

Summary:
X-link: facebookresearch/FBGEMM#978

Pull Request resolved: #3886

-  Add FB python sources into genai CMakeLists.txt

Differential Revision: D71930483

Author: q10

fbgemm_gpu/experimental/gen_ai/CMakeLists.txt

@@ -26,14 +26,6 @@ file(GLOB experimental_gen_ai_cpp_source_files_gpu
   src/quantize/*.cpp
   src/quantize/*.cu)
 
-# Set the source file for FB only CPP
-if(USE_FB_ONLY)
-  file(GLOB fb_only_ops_sources
-      fb/src/*/*.cu
-      fb/src/*/*.cpp)
-  list(APPEND experimental_gen_ai_cpp_source_files_gpu ${fb_only_ops_sources})
-endif()
-
 # CUDA-specific sources
 file(GLOB_RECURSE experimental_gen_ai_cpp_source_files_cuda
   src/quantize/cutlass_extensions/*.cu
@@ -53,6 +45,14 @@ file(GLOB_RECURSE experimental_gen_ai_cpp_source_files_hip
 file(GLOB_RECURSE experimental_gen_ai_python_source_files
   gen_ai/*.py)
 
+  # Set the source file for FB only CPP
+if(USE_FB_ONLY)
+  file(GLOB fb_only_cxx_sources
+      fb/src/*/*.cu
+      fb/src/*/*.cpp)
+  list(APPEND experimental_gen_ai_cpp_source_files_gpu ${fb_only_cxx_sources})
+endif()
+
 
 ################################################################################
 # Build Shared Library
@@ -82,3 +82,9 @@ add_to_package(
   DESTINATION fbgemm_gpu/experimental/gen_ai
   TARGETS fbgemm_gpu_experimental_gen_ai
   FILES ${experimental_gen_ai_python_source_files})
+
+if(USE_FB_ONLY)
+  install(
+    DIRECTORY fb/gen_ai
+    DESTINATION fbgemm_gpu/experimental/gen_ai)
+endif()