diff --git a/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt b/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt index 6687f1e6a6..315ef5a889 100644 --- a/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt +++ b/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt @@ -26,7 +26,7 @@ set(CMAKE_VERBOSE_MAKEFILE ON) # Target Sources ################################################################################ -glob_files_nohip(experimental_gen_ai_cpp_source_files_cpu +glob_files(experimental_gen_ai_cpp_source_files_cpu src/attention/cuda/*.cpp src/coalesce/*.cpp src/comm/*.cpp @@ -35,7 +35,7 @@ glob_files_nohip(experimental_gen_ai_cpp_source_files_cpu src/quantize/*.cpp src/quantize/common/*.cpp) -glob_files_nohip(experimental_gen_ai_cpp_source_files_gpu +glob_files(experimental_gen_ai_cpp_source_files_gpu src/attention/cuda/*.cu src/coalesce/*.cu src/comm/*.cu