(tvm) branch main updated: [CMake] Add "USE_FLASHINFER" to libinfo (#16710)

[syfeng]

This is an automated email from the ASF dual-hosted git repository.

syfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 6a877df173 [CMake] Add "USE_FLASHINFER" to libinfo (#16710)
6a877df173 is described below

commit 6a877df173093480f57b8c1e2b199ff01865b49a
Author: Ruihang Lai <ruiha...@cs.cmu.edu>
AuthorDate: Wed Mar 13 05:51:00 2024 -0400

    [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.
---
 cmake/modules/LibInfo.cmake | 1 +
 src/support/libinfo.cc      | 1 +
 2 files changed, 2 insertions(+)

diff --git a/cmake/modules/LibInfo.cmake b/cmake/modules/LibInfo.cmake
index b971919acf..5f82a0c782 100644
--- 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
index cc84f7a675..38159c42eb 100644
--- 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},