This PR does the following: - CUDA utilities (e.g., cuGetInfo) won't be compiled as part of libtriton.so anymore. - Refactoring driver/llvm.cc to split it between PTX codegen and python. - By extension this will also deprecate include/external so Triton won't have to live with a copy of some CUDA/Hip headers anymore. - `triton-translate` becomes a `triton.tools.aot` Python utility that re-uses functions from the triton.compile sub-module.
61 lines
1.5 KiB
CMake
61 lines
1.5 KiB
CMake
add_subdirectory(FileCheck)
|
|
# add_llvm_executable(FileCheck FileCheck/FileCheck.cpp)
|
|
# target_link_libraries(FileCheck PRIVATE LLVMFileCheck LLVMSupport)
|
|
|
|
get_property(dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)
|
|
get_property(conversion_libs GLOBAL PROPERTY MLIR_CONVERSION_LIBS)
|
|
|
|
add_llvm_executable(triton-opt triton-opt.cpp PARTIAL_SOURCES_INTENDED)
|
|
|
|
# TODO: what's this?
|
|
llvm_update_compile_flags(triton-opt)
|
|
target_link_libraries(triton-opt PRIVATE
|
|
TritonAnalysis
|
|
TritonTransforms
|
|
TritonGPUTransforms
|
|
${dialect_libs}
|
|
${conversion_libs}
|
|
# tests
|
|
TritonTestAnalysis
|
|
# MLIR core
|
|
MLIROptLib
|
|
MLIRPass
|
|
MLIRTransforms
|
|
)
|
|
|
|
mlir_check_all_link_libraries(triton-opt)
|
|
|
|
|
|
# add_llvm_executable(triton-translate triton-translate.cpp PARTIAL_SOURCES_INTENDED)
|
|
#llvm_update_compile_flags(triton-translate)
|
|
# target_link_libraries(triton-translate PRIVATE
|
|
# TritonAnalysis
|
|
# TritonTransforms
|
|
# TritonGPUTransforms
|
|
# TritonLLVMIR
|
|
# TritonDriver
|
|
# ${dialect_libs}
|
|
# ${conversion_libs}
|
|
# # tests
|
|
# TritonTestAnalysis
|
|
|
|
# LLVMCore
|
|
# LLVMSupport
|
|
# LLVMOption
|
|
# LLVMCodeGen
|
|
# LLVMAsmParser
|
|
|
|
# # MLIR core
|
|
# MLIROptLib
|
|
# MLIRIR
|
|
# MLIRPass
|
|
# MLIRSupport
|
|
# MLIRTransforms
|
|
# MLIRExecutionEngine
|
|
# MLIRMathToLLVM
|
|
# MLIRTransformUtils
|
|
# MLIRLLVMToLLVMIRTranslation
|
|
# MLIRNVVMToLLVMIRTranslation
|
|
# )
|
|
# mlir_check_all_link_libraries(triton-translate)
|