experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allow MLIR external repository to be created from non-tensorflow repository.

Browse Source 
PiperOrigin-RevId: 268519854
This commit is contained in:
Christian Sigg 2019-09-11 12:43:35 -07:00 committed by minglotus-6
parent 44e5d979ca
commit af33d31eca
4 changed files with 25 additions and 5 deletions

2
tensorflow/opensource_only.files
View File

@ -1,4 +1,4 @@
llvm/llvm/projects/google_mlir/WORKSPACE
llvm/llvm/projects/google_mlir/mlir_configure.bzl
tensorflow/__init__.py
tensorflow/api_template.__init__.py
tensorflow/api_template_v1.__init__.py

6
tensorflow/workspace.bzl
View File

@ -7,6 +7,7 @@ load("//third_party/nccl:nccl_configure.bzl", "nccl_configure")
load("//third_party/mkl:build_defs.bzl", "mkl_repository")
load("//third_party/git:git_configure.bzl", "git_configure")
load("//third_party/py:python_configure.bzl", "python_configure")
load("//third_party/mlir:mlir_configure.bzl", "mlir_configure")
load("//third_party/sycl:sycl_configure.bzl", "sycl_configure")
load("//third_party/systemlibs:syslibs_configure.bzl", "syslibs_configure")
load("//third_party/toolchains/remote:configure.bzl", "remote_execution_configure")
@ -75,10 +76,7 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
syslibs_configure(name = "local_config_syslibs")
python_configure(name = "local_config_python")
rocm_configure(name = "local_config_rocm")
native.local_repository(
name = "local_config_mlir",
path = "third_party/mlir",
)
mlir_configure(name = "local_config_mlir")
remote_execution_configure(name = "local_config_remote_execution")
initialize_third_party()

0
third_party/mlir/WORKSPACE vendored
View File

22
third_party/mlir/mlir_configure.bzl vendored Normal file
View File

@ -0,0 +1,22 @@
"""Repository rule for MLIR autoconfiguration."""
def _mlir_configure_impl(repository_ctx):
repository_ctx.file("WORKSPACE", "")
label = Label("@org_tensorflow//third_party/mlir:mlir_configure.bzl")
for entry in repository_ctx.path(label).dirname.readdir():
repository_ctx.symlink(entry, entry.basename)
mlir_configure = repository_rule(
implementation = _mlir_configure_impl,
)
"""Detects and configures the MLIR configuration.
Add the following to your WORKSPACE FILE:
```python
mlir_configure(name = "local_config_mlir")
```
Args:
name: A unique name for this workspace rule.
"""