Allow ONNX and TF modules optional

CMakeLists.txt

@@ -87,8 +87,10 @@ endif()
 
 option(MIGRAPHX_USE_ROCBLAS "Enable MIGraphX to use rocBLAS" ON)
 option(MIGRAPHX_USE_HIPBLASLT "Enable MIGraphX to use hipBLASLt" ON)
-option(MIGRAPHX_ENABLE_PYTHON "Enable python bindings" ON)
 option(MIGRAPHX_USE_MIOPEN "Enable MIGraphX to use MIOpen" ON)
+option(MIGRAPHX_ENABLE_PYTHON "Enable python bindings" ON)
+option(MIGRAPHX_ENABLE_TENSORFLOW "Enable TensorFlow integration" ON)
+option(MIGRAPHX_ENABLE_ONNX "Enable ONNX model import" ON)
 
 if(MIGRAPHX_USE_HIPBLASLT AND NOT MIGRAPHX_USE_ROCBLAS)
     message(FATAL_ERROR "hipBLASLt requires rocBLAS, but MIGRAPHX_USE_ROCBLAS is OFF")
@@ -406,7 +408,6 @@ if(BUILD_TESTING)
     rocm_enable_test_package(migraphx)
     add_subdirectory(test)
 endif()
-add_subdirectory(tools)
 
 set(DEST_DIR ${CMAKE_BINARY_DIR})
 file(GLOB backend_files ${CMAKE_SOURCE_DIR}/src/py/backend/*.py)

src/CMakeLists.txt

@@ -163,6 +163,8 @@ if(WIN32)
     target_compile_options(migraphx PUBLIC "SHELL:-Xclang -cfguard")
 endif()
 
+configure_file(config.h.in include/migraphx/config.h)
+
 configure_file(version.h.in include/migraphx/version.h)
 add_library(migraphx_version INTERFACE)
 rocm_install_targets(
@@ -398,12 +400,18 @@ target_link_libraries(migraphx PRIVATE msgpack-cxx)
 target_link_libraries(migraphx INTERFACE $<BUILD_INTERFACE:msgpack-cxx>)
 
 add_library(migraphx_all_targets INTERFACE)
+add_library(migraphx_all_frontends INTERFACE)
 
 add_subdirectory(api)
 add_subdirectory(driver)
+if(MIGRAPHX_ENABLE_ONNX)
 add_subdirectory(onnx)
+target_link_libraries(migraphx_all_frontends INTERFACE migraphx_onnx)
+endif()
+if(MIGRAPHX_ENABLE_TENSORFLOW)
 add_subdirectory(tf)
-
+target_link_libraries(migraphx_all_frontends INTERFACE migraphx_tf)
+endif()
 if(MIGRAPHX_ENABLE_PYTHON)
 add_subdirectory(py)
 endif()
@@ -445,6 +453,7 @@ if(BUILD_DEV)
 endif()
 
 target_compile_definitions(migraphx PUBLIC MIGRAPHX_CXX_COMPILER="${CMAKE_CXX_COMPILER}")
+add_dependencies(migraphx generate)
 
 rocm_export_targets(
   TARGETS migraphx::migraphx_c

src/api/CMakeLists.txt

@@ -1,7 +1,7 @@
 #####################################################################################
 # The MIT License (MIT)
 #
-# Copyright (c) 2015-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright (c) 2015-2026 Advanced Micro Devices, Inc. All rights reserved.
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -22,9 +22,37 @@
 # THE SOFTWARE.
 #####################################################################################
 
-add_library(migraphx_c
-    api.cpp
-)
+find_package(Python 3 REQUIRED COMPONENTS Interpreter)
+find_program(CLANG_FORMAT clang-format PATHS /opt/rocm/llvm ENV HIP_PATH PATH_SUFFIXES bin)
+
+set(TOOLS_DIR "${PROJECT_SOURCE_DIR}/tools")
+set(BIN_DIR "${PROJECT_BINARY_DIR}/src/api")
+set(SRC_DIR "${PROJECT_SOURCE_DIR}/src/api")
+
+set(GENERATE_COMMAND ${Python_EXECUTABLE} generate.py)
+if (CLANG_FORMAT)
+    list(APPEND GENERATE_COMMAND -f ${CLANG_FORMAT})
+endif()
+if(MIGRAPHX_ENABLE_ONNX)
+    list(APPEND GENERATE_COMMAND -Denable_onnx)
+endif()
+if(MIGRAPHX_ENABLE_TENSORFLOW)
+    list(APPEND GENERATE_COMMAND -Denable_tensorflow)
+endif()
+list(APPEND GENERATE_COMMAND -o ${PROJECT_BINARY_DIR}/src)
+
+add_custom_command(
+    OUTPUT ${BIN_DIR}/api.cpp
+    COMMAND ${GENERATE_COMMAND}
+    DEPENDS ${TOOLS_DIR}/generate.py ${TOOLS_DIR}/api.py ${TOOLS_DIR}/te.py ${SRC_DIR}/migraphx.py
+    WORKING_DIRECTORY ${TOOLS_DIR}
+    COMMENT "Generate MIGraphX C API source files...")
+
+add_custom_target(generate DEPENDS ${BIN_DIR}/api.cpp)
+
+add_library(migraphx_c ${BIN_DIR}/api.cpp)
+add_dependencies(migraphx_c generate)
+
 set_target_properties(migraphx_c PROPERTIES EXPORT_NAME c)
 migraphx_generate_export_header(migraphx_c DIRECTORY migraphx/api)
 
@@ -37,7 +65,7 @@ if(BUILD_TESTING)
 endif()
 
 rocm_clang_tidy_check(migraphx_c)
-target_link_libraries(migraphx_c PRIVATE migraphx migraphx_tf migraphx_onnx)
+target_link_libraries(migraphx_c PRIVATE migraphx migraphx_all_frontends)
 target_link_libraries(migraphx_c PUBLIC migraphx_version)
 
 rocm_install_targets(