[CMake] Add "USE_FLASHINFER" to libinfo (#16710)

This PR adds the flag `USE_FLASHINFER` to libinfo, so that we can use the global function "support.GetLibInfo" to check if FlashInfer is enabled when building TVM.
apache · Mar 13, 2024 · 6a877df · 6a877df
1 parent 3b25697
commit 6a877df
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/cmake/modules/LibInfo.cmake b/cmake/modules/LibInfo.cmake
@@ -75,6 +75,7 @@ function(add_lib_info src_file)
     TVM_INFO_USE_CUDNN="${USE_CUDNN}"
     TVM_INFO_USE_CUSTOM_LOGGING="${USE_CUSTOM_LOGGING}"
     TVM_INFO_USE_CUTLASS="${USE_CUTLASS}"
+    TVM_INFO_USE_FLASHINFER="${USE_FLASHINFER}"
     TVM_INFO_USE_AMX="${USE_AMX}"
     TVM_INFO_USE_DNNL="${USE_DNNL}"
     TVM_INFO_USE_ETHOSN="${USE_ETHOSN}"

diff --git a/src/support/libinfo.cc b/src/support/libinfo.cc
@@ -307,6 +307,7 @@ TVM_DLL Map<String, String> GetLibInfo() {
       {"USE_CUDNN", TVM_INFO_USE_CUDNN},
       {"USE_CUSTOM_LOGGING", TVM_INFO_USE_CUSTOM_LOGGING},
       {"USE_CUTLASS", TVM_INFO_USE_CUTLASS},
+      {"USE_FLASHINFER", TVM_INFO_USE_FLASHINFER},
       {"USE_AMX", TVM_INFO_USE_AMX},
       {"USE_DNNL", TVM_INFO_USE_DNNL},
       {"USE_ETHOSN", TVM_INFO_USE_ETHOSN},