diff --git a/lib/Analysis/CMakeLists.txt b/lib/Analysis/CMakeLists.txt index 1f761f845cd9..bb9a21ccf980 100644 --- a/lib/Analysis/CMakeLists.txt +++ b/lib/Analysis/CMakeLists.txt @@ -11,4 +11,7 @@ add_mlir_library(TritonAnalysis LINK_LIBS PUBLIC MLIRAnalysis + MLIRLLVMDialect + TritonIR + TritonGPUIR ) diff --git a/lib/Conversion/TritonToTritonGPU/CMakeLists.txt b/lib/Conversion/TritonToTritonGPU/CMakeLists.txt index 9d7f961cccf2..834d10a4dea0 100644 --- a/lib/Conversion/TritonToTritonGPU/CMakeLists.txt +++ b/lib/Conversion/TritonToTritonGPU/CMakeLists.txt @@ -14,6 +14,7 @@ add_mlir_conversion_library(TritonToTritonGPU LINK_LIBS PUBLIC MLIRIR MLIRPass + MLIRTransforms TritonIR TritonGPUIR TritonGPUTransforms diff --git a/lib/Dialect/Triton/IR/CMakeLists.txt b/lib/Dialect/Triton/IR/CMakeLists.txt index 819bdfcfb83b..6ee110718c6d 100644 --- a/lib/Dialect/Triton/IR/CMakeLists.txt +++ b/lib/Dialect/Triton/IR/CMakeLists.txt @@ -10,5 +10,6 @@ add_mlir_dialect_library(TritonIR LINK_LIBS PUBLIC MLIRIR MLIRArithDialect + MLIRMathDialect MLIRSCFDialect ) diff --git a/lib/Dialect/Triton/Transforms/CMakeLists.txt b/lib/Dialect/Triton/Transforms/CMakeLists.txt index b6e3b1f54e9f..dcd050a459fe 100644 --- a/lib/Dialect/Triton/Transforms/CMakeLists.txt +++ b/lib/Dialect/Triton/Transforms/CMakeLists.txt @@ -9,4 +9,9 @@ add_mlir_dialect_library(TritonTransforms DEPENDS TritonTransformsIncGen TritonCombineIncGen + + LINK_LIBS PUBLIC + MLIRPass + MLIRTransformUtils + TritonIR ) diff --git a/lib/Dialect/TritonGPU/IR/CMakeLists.txt b/lib/Dialect/TritonGPU/IR/CMakeLists.txt index 903dfc318467..20f6f9851e1b 100644 --- a/lib/Dialect/TritonGPU/IR/CMakeLists.txt +++ b/lib/Dialect/TritonGPU/IR/CMakeLists.txt @@ -7,5 +7,6 @@ add_mlir_dialect_library(TritonGPUIR TritonGPUAttrDefsIncGen LINK_LIBS PUBLIC + MLIRGPUOps TritonIR ) diff --git a/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt b/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt index 37fbfb0463a1..ced082069198 100644 --- a/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt +++ b/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt @@ -14,7 +14,9 @@ add_mlir_dialect_library(TritonGPUTransforms TritonGPUTransformsIncGen LINK_LIBS PUBLIC + MLIRTransforms + MLIRTransformUtils + TritonAnalysis TritonIR TritonGPUIR - MLIRTransformUtils ) diff --git a/lib/Target/LLVMIR/CMakeLists.txt b/lib/Target/LLVMIR/CMakeLists.txt index ac8973ad1960..1430a6af223c 100644 --- a/lib/Target/LLVMIR/CMakeLists.txt +++ b/lib/Target/LLVMIR/CMakeLists.txt @@ -5,8 +5,17 @@ add_mlir_translation_library(TritonLLVMIR Core LINK_LIBS PUBLIC + MLIRArithToLLVM + MLIRBuiltinToLLVMIRTranslation + MLIRExecutionEngineUtils + MLIRIndexToLLVM MLIRIR MLIRLLVMDialect + MLIRLLVMToLLVMIRTranslation + MLIRNVVMToLLVMIRTranslation + MLIRROCDLToLLVMIRTranslation + MLIRSCFToControlFlow MLIRSupport MLIRTargetLLVMIRExport + TritonGPUToLLVM ) diff --git a/test/lib/Analysis/CMakeLists.txt b/test/lib/Analysis/CMakeLists.txt index deafc7b8adef..e3c7743795c4 100644 --- a/test/lib/Analysis/CMakeLists.txt +++ b/test/lib/Analysis/CMakeLists.txt @@ -5,6 +5,7 @@ add_mlir_library(TritonTestAnalysis TestMembar.cpp LINK_LIBS PUBLIC + MLIRPass TritonAnalysis ${dialect_libs} ) diff --git a/unittest/Analysis/CMakeLists.txt b/unittest/Analysis/CMakeLists.txt index 880c8117b162..829d0bff7bf6 100644 --- a/unittest/Analysis/CMakeLists.txt +++ b/unittest/Analysis/CMakeLists.txt @@ -1,5 +1,8 @@ add_triton_ut( NAME TestTritonAnalysis SRCS UtilityTest.cpp - LIBS TritonAnalysis + LIBS + TritonAnalysis + TritonIR + TritonGPUIR )