From f3c33871861234e508c75cc1397046708adba089 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 9 Apr 2020 20:14:21 -0700
Subject: [PATCH] Internal change
PiperOrigin-RevId: 305813517
Change-Id: I685ddcfb9861dab938bf5d46ad3d0874f04590bc
---
third_party/mlir/BUILD | 22 ++++++++++++----------
third_party/mlir/test.BUILD | 17 +++++++++++++++++
2 files changed, 29 insertions(+), 10 deletions(-)
diff --git a/third_party/mlir/BUILD b/third_party/mlir/BUILD
index a8e41daffe3..a9199ba9505 100644
--- a/third_party/mlir/BUILD
+++ b/third_party/mlir/BUILD
@@ -342,7 +342,6 @@ cc_library(
":Transforms",
"@llvm-project//llvm:support",
],
- alwayslink = 1,
)
filegroup(
@@ -850,7 +849,6 @@ cc_library(
"@llvm-project//llvm:core",
"@llvm-project//llvm:support",
],
- alwayslink = 1,
)
cc_library(
@@ -1770,7 +1768,6 @@ cc_library(
":Translation",
"@llvm-project//llvm:support",
],
- alwayslink = 1,
)
cc_library(
@@ -2251,7 +2248,6 @@ cc_library(
"@llvm-project//llvm:ir_reader",
"@llvm-project//llvm:support",
],
- alwayslink = 1,
)
cc_library(
@@ -2275,7 +2271,6 @@ cc_library(
"@llvm-project//llvm:core",
"@llvm-project//llvm:support",
],
- alwayslink = 1,
)
cc_library(
@@ -2299,7 +2294,6 @@ cc_library(
"@llvm-project//llvm:core",
"@llvm-project//llvm:support",
],
- alwayslink = 1,
)
# TODO(zinenko): Update these so that we can simplify mapping to cmake.
@@ -2388,11 +2382,23 @@ cc_library(
],
)
+cc_library(
+ name = "AllTranslations",
+ hdrs = ["include/mlir/InitAllTranslations.h"],
+ deps = [
+ ":SPIRVTranslateRegistration",
+ ":TargetLLVMIR",
+ ":TargetNVVMIR",
+ ":TargetROCDLIR",
+ ],
+)
+
cc_library(
name = "MlirTranslateMain",
srcs = ["tools/mlir-translate/mlir-translate.cpp"],
deps = [
":AllPassesAndDialectsNoRegistration",
+ ":AllTranslations",
":IR",
":Parser",
":Support",
@@ -2405,10 +2411,6 @@ cc_binary(
name = "mlir-translate",
deps = [
":MlirTranslateMain",
- ":SPIRVTranslateRegistration",
- ":TargetLLVMIR",
- ":TargetNVVMIR",
- ":TargetROCDLIR",
],
)
diff --git a/third_party/mlir/test.BUILD b/third_party/mlir/test.BUILD
index 476fd8b77df..f242ae76287 100644
--- a/third_party/mlir/test.BUILD
+++ b/third_party/mlir/test.BUILD
@@ -46,6 +46,22 @@ gentbl(
],
)
+gentbl(
+ name = "TestLinalgMatmulToVectorPatternsIncGen",
+ tbl_outs = [
+ (
+ "-gen-rewriters",
+ "lib/DeclarativeTransforms/TestLinalgMatmulToVectorPatterns.h.inc",
+ ),
+ ],
+ tblgen = "@llvm-project//mlir:mlir-tblgen",
+ td_file = "lib/DeclarativeTransforms/TestLinalgMatmulToVectorPatterns.td",
+ td_srcs = [
+ "@llvm-project//mlir:VectorTransformPatternsTdFiles",
+ "@llvm-project//mlir:LinalgTransformPatternsTdFiles",
+ ],
+)
+
gentbl(
name = "TestOpsIncGen",
strip_include_prefix = "lib/Dialect/Test",
@@ -157,6 +173,7 @@ cc_library(
includes = ["lib/Dialect/Test"],
deps = [
":TestDialect",
+ ":TestLinalgMatmulToVectorPatternsIncGen",
":TestLinalgTransformPatternsIncGen",
":TestVectorTransformPatternsIncGen",
"@llvm-project//llvm:support",