CMakeLists.txt 390 字节
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
if(((NOT WITH_GPU) AND (NOT WITH_ROCM)) OR WIN32 OR APPLE)
5
  LIST(REMOVE_ITEM TEST_IR_PASSES test_ir_fusion_group_pass)
6 7
endif()

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

add_subdirectory(inference)