156 lines
3.0 KiB
CMake
156 lines
3.0 KiB
CMake
|
#
|
||
|
|
# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# https://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
#
|
||
|
|
include_directories(BEFORE
|
||
|
|
${CMAKE_CURRENT_BINARY_DIR}
|
||
|
|
${CMAKE_CURRENT_SOURCE_DIR})
|
||
|
|
|
||
|
|
set(LLVM_TARGET_DEFINITIONS lower_complex_patterns.td)
|
||
|
|
mlir_tablegen(generated_lower_complex.inc -gen-rewriters)
|
||
|
|
add_public_tablegen_target(MLIRMhloLowerComplexIncGen)
|
||
|
|
|
||
|
|
set(LLVM_TARGET_DEFINITIONS legalize_to_standard_patterns.td)
|
||
|
|
mlir_tablegen(generated_legalize_to_standard.inc -gen-rewriters)
|
||
|
|
add_public_tablegen_target(MLIRMhloLegalizeToStandardIncGen)
|
||
|
|
|
||
|
|
|
||
|
|
add_mlir_library(ChloPasses
|
||
|
|
chlo_legalize_to_hlo.cc
|
||
|
|
chlo_legalize_to_hlo_pass.cc
|
||
|
|
|
||
|
|
DEPENDS
|
||
|
|
MLIRhlo_opsIncGen
|
||
|
|
|
||
|
|
LINK_COMPONENTS
|
||
|
|
Core
|
||
|
|
|
||
|
|
LINK_LIBS PUBLIC
|
||
|
|
ChloDialect
|
||
|
|
MLIRIR
|
||
|
|
MLIRPass
|
||
|
|
)
|
||
|
|
|
||
|
|
add_mlir_library(MhloPasses
|
||
|
|
legalize_gather_to_torch_index_select.cc
|
||
|
|
legalize_tanh_to_approximation.cc
|
||
|
|
lower_complex.cc
|
||
|
|
lower_complex_patterns.td
|
||
|
|
lower_general_dot.cc
|
||
|
|
materialize_broadcasts.cc
|
||
|
|
materialize_broadcasts_pass.cc
|
||
|
|
mhlo_fusion.cc
|
||
|
|
optimize_mhlo.cc
|
||
|
|
optimize_mhlo_pass.cc
|
||
|
|
sink_constants_to_control_flow.cc
|
||
|
|
test_infer_shaped_type_pass.cc
|
||
|
|
transform_unranked_hlo.cc
|
||
|
|
unfuse_batch_norm.cc
|
||
|
|
unfuse_batch_norm_pass.cc
|
||
|
|
|
||
|
|
DEPENDS
|
||
|
|
MLIRhlo_opsIncGen
|
||
|
|
MLIRMhloLowerComplexIncGen
|
||
|
|
|
||
|
|
LINK_COMPONENTS
|
||
|
|
Core
|
||
|
|
|
||
|
|
LINK_LIBS PUBLIC
|
||
|
|
MLIRIR
|
||
|
|
MLIRMhloUtils
|
||
|
|
MLIRPass
|
||
|
|
MLIRTransformUtils
|
||
|
|
)
|
||
|
|
|
||
|
|
add_mlir_library(MhloToLhloConversion
|
||
|
|
hlo_legalize_to_lhlo.cc
|
||
|
|
|
||
|
|
DEPENDS
|
||
|
|
MLIRhlo_opsIncGen
|
||
|
|
MLIRlhlo_opsIncGen
|
||
|
|
|
||
|
|
LINK_COMPONENTS
|
||
|
|
Core
|
||
|
|
|
||
|
|
LINK_LIBS PUBLIC
|
||
|
|
MhloDialect
|
||
|
|
LmhloDialect
|
||
|
|
MLIRIR
|
||
|
|
MLIRPass
|
||
|
|
)
|
||
|
|
|
||
|
|
add_mlir_library(MhloToStandard
|
||
|
|
legalize_control_flow.cc
|
||
|
|
legalize_to_standard.cc
|
||
|
|
|
||
|
|
DEPENDS
|
||
|
|
MLIRhlo_opsIncGen
|
||
|
|
MLIRlhlo_opsIncGen
|
||
|
|
MLIRMhloLegalizeToStandardIncGen
|
||
|
|
|
||
|
|
LINK_COMPONENTS
|
||
|
|
Core
|
||
|
|
|
||
|
|
LINK_LIBS PUBLIC
|
||
|
|
MLIRIR
|
||
|
|
MLIRPass
|
||
|
|
)
|
||
|
|
|
||
|
|
add_mlir_library(MhloLhloToLinalg
|
||
|
|
legalize_to_linalg.cc
|
||
|
|
|
||
|
|
DEPENDS
|
||
|
|
MLIRhlo_opsIncGen
|
||
|
|
MLIRlhlo_opsIncGen
|
||
|
|
|
||
|
|
LINK_COMPONENTS
|
||
|
|
Core
|
||
|
|
|
||
|
|
LINK_LIBS PUBLIC
|
||
|
|
MhloDialect
|
||
|
|
MLIRIR
|
||
|
|
MLIRPass
|
||
|
|
)
|
||
|
|
|
||
|
|
add_mlir_library(LmhloPasses
|
||
|
|
lhlo_copy_removal.cc
|
||
|
|
lhlo_fuse_linalg.cc
|
||
|
|
lhlo_legalize_to_affine.cc
|
||
|
|
lhlo_legalize_to_gpu.cc
|
||
|
|
lhlo_legalize_to_llvm.cc
|
||
|
|
lhlo_legalize_to_llvm_pass.cc
|
||
|
|
lhlo_legalize_to_parallel_loops.cc
|
||
|
|
|
||
|
|
DEPENDS
|
||
|
|
MLIRlhlo_opsIncGen
|
||
|
|
|
||
|
|
LINK_COMPONENTS
|
||
|
|
Core
|
||
|
|
|
||
|
|
LINK_LIBS PUBLIC
|
||
|
|
LmhloDialect
|
||
|
|
MLIRIR
|
||
|
|
MLIRPass
|
||
|
|
)
|
||
|
|
|
||
|
|
add_library(AllMhloPasses INTERFACE)
|
||
|
|
target_link_libraries(AllMhloPasses INTERFACE
|
||
|
|
ChloPasses
|
||
|
|
MhloPasses
|
||
|
|
MhloToLhloConversion
|
||
|
|
MhloToStandard
|
||
|
|
MhloLhloToLinalg
|
||
|
|
LmhloPasses
|
||
|
|
)
|