[fbgemm_gpu] Fix CMakeLists.txt for experimental/gemm

- Fix CMakeLists.txt for experimental/gemm
pytorch · Jan 21, 2025 · cbf53e2 · cbf53e2
1 parent 2d025dc
commit cbf53e2
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/fbgemm_gpu/experimental/gemm/CMakeLists.txt b/fbgemm_gpu/experimental/gemm/CMakeLists.txt
@@ -8,15 +8,20 @@
 # 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
+  RELATIVE 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
+  FILES ${experimental_triton_gemm_python_source_files})