From 919d03a89b250a6cb9c97c40212f739990c85ee6 Mon Sep 17 00:00:00 2001
From: Ruihang Lai <ruihangl@cs.cmu.edu>
Date: Tue, 12 Mar 2024 13:36:56 -0400
Subject: [PATCH] [CMake] Add "USE_FLASHINFER" to libinfo

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 b971919acf23..5f82a0c78286 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 cc84f7a6755b..38159c42ebd3 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},