diff --git a/test/lit.cfg.py b/test/lit.cfg.py index 3517a37249..a406eefa3e 100644 --- a/test/lit.cfg.py +++ b/test/lit.cfg.py @@ -45,7 +45,6 @@ # test_exec_root: The root path where tests should be run. config.test_exec_root = os.path.join(config.triton_obj_root, 'test') config.triton_tools_dir = os.path.join(config.triton_obj_root, 'bin') -config.triton_intel_plugin_tools_dir = os.path.join(config.triton_obj_root, 'third_party', 'intel', 'bin') config.filecheck_dir = os.path.join(config.triton_obj_root, 'bin', 'FileCheck') # FileCheck -enable-var-scope is enabled by default in MLIR test @@ -53,7 +52,7 @@ # it can be explicitly opted-in by prefixing the variable name with $ config.environment["FILECHECK_OPTS"] = "--enable-var-scope" -tool_dirs = [config.triton_tools_dir, config.triton_intel_plugin_tools_dir, config.llvm_tools_dir, config.filecheck_dir] +tool_dirs = [config.triton_tools_dir, config.llvm_tools_dir, config.filecheck_dir] # Tweak the PATH to include the tools dir. for d in tool_dirs: diff --git a/third_party/intel/bin/CMakeLists.txt b/third_party/intel/bin/CMakeLists.txt index 1054f32eb3..2f5925b056 100644 --- a/third_party/intel/bin/CMakeLists.txt +++ b/third_party/intel/bin/CMakeLists.txt @@ -22,3 +22,10 @@ target_link_libraries(triton-translate TritonGENToLLVMIRTranslation ) mlir_check_link_libraries(triton-translate) + +# Move `triton-translate` to the folder where `lit` expects to find it +add_custom_command(TARGET triton-translate POST_BUILD + COMMAND ${CMAKE_COMMAND} -E copy + "$" + "${triton_BINARY_DIR}/bin" +)