[fbgemm_gpu] Fix CMakeLists.txt for experimental/gemm

- Fix CMakeLists.txt for experimental/gemm
pytorch · Jan 21, 2025 · 53b7dbf · 53b7dbf
1 parent 2d025dc
commit 53b7dbf
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/fbgemm_gpu/experimental/gemm/CMakeLists.txt b/fbgemm_gpu/experimental/gemm/CMakeLists.txt
@@ -8,15 +8,19 @@
 # Target Sources
 ################################################################################
 
-set(experimental_triton_python_source_files
-    triton_gemm/__init__.py
-    triton_gemm/fp8_gemm.py
-    triton_gemm/matmul_perf_model.py)
+# Python sources
+file(GLOB_RECURSE experimental_triton_gemm_python_source_files
+  triton_gemm/*.py)
 
+BLOCK_PRINT(
+  "Globbed GEMM FILES"
+  "${experimental_triton_gemm_python_source_files}"
+)
 
 ################################################################################
 # Install Python Files
 ################################################################################
 
-add_to_package(DESTINATION fbgemm_gpu/experimental/gemm/triton_gemm
-    FILES ${experimental_triton_python_source_files})
+add_to_package(
+  DESTINATION fbgemm_gpu/experimental/gemm/triton_gemm
+  FILES ${experimental_triton_gemm_python_source_files})
diff --git a/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt b/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt
@@ -42,8 +42,12 @@ file(GLOB_RECURSE experimental_gen_ai_cpp_source_files_hip
 
 # Python sources
 file(GLOB_RECURSE experimental_gen_ai_python_source_files
-  RELATIVE gen_ai
-  *.py)
+  gen_ai/*.py)
+
+  BLOCK_PRINT(
+    "Globbed GEN AI FILES"
+    "${experimental_gen_ai_python_source_files}"
+  )
 
 
 ################################################################################