wip

pytorch · May 25, 2024 · cd1f7c0 · cd1f7c0
1 parent 6deac62
commit cd1f7c0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cmake/modules/PyTorchSetup.cmake b/cmake/modules/PyTorchSetup.cmake
@@ -17,6 +17,7 @@ BLOCK_PRINT(
   "PyTorch Flags"
   ""
   "TORCH_INCLUDE_DIRS=${TORCH_INCLUDE_DIRS}"
+  ""
   "TORCH_LIBRARIES=${TORCH_LIBRARIES}"
 )
 

diff --git a/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt b/fbgemm_gpu/experimental/gen_ai/CMakeLists.txt
@@ -8,6 +8,15 @@
 # Target Sources
 ################################################################################
 
+BLOCK_PRINT(
+  "PyTorch Flags Inside GenAI"
+  ""
+  "TORCH_INCLUDE_DIRS=${TORCH_INCLUDE_DIRS}"
+  ""
+  "TORCH_LIBRARIES=${TORCH_LIBRARIES}"
+)
+
+
 if(FBGEMM_GENAI_ONLY)
   set(fbgemm_sources_include_directories
     # FBGEMM