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 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},