CMakeLists.txt 361 字节
Newer Older
1 2 3
file(GLOB TEST_IR_PASSES RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_IR_PASSES "${TEST_IR_PASSES}")

4 5 6 7
if(NOT WITH_GPU OR WIN32 OR APPLE)
  LIST(REMOVE_ITEM TEST_IR_PASSES test_ir_fusion_group)
endif()

8 9 10
foreach(target ${TEST_IR_PASSES})
  py_test_modules(${target} MODULES ${target})
endforeach()
11 12

add_subdirectory(inference)