forked from DeepRec-AI/DeepRec
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #6758 from drpngx/branch_144040081
Branch 144040081
- Loading branch information
Showing
738 changed files
with
143,589 additions
and
3,246 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,218 @@ | ||
| licenses(["notice"]) # Apache 2.0 | ||
|
|
||
| package( | ||
| default_visibility = ["//visibility:private"], | ||
| ) | ||
|
|
||
| load("//tensorflow/compiler/xla:xla.bzl", "xla_proto_library") | ||
| load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library") | ||
|
|
||
| # Optional runtime utilities for use by code generated by tfcompile. | ||
| cc_library( | ||
| name = "runtime", | ||
| srcs = ["runtime.cc"], | ||
| hdrs = ["runtime.h"], | ||
| visibility = ["//visibility:public"], | ||
| deps = [ | ||
| "//tensorflow/core:framework_lite", | ||
| ], | ||
| ) | ||
|
|
||
| cc_test( | ||
| name = "runtime_test", | ||
| srcs = ["runtime_test.cc"], | ||
| deps = [ | ||
| ":runtime", | ||
| "//tensorflow/compiler/tf2xla:xla_local_runtime_context", | ||
| "//tensorflow/core:framework", | ||
| "//tensorflow/core:test", | ||
| "//tensorflow/core:test_main", | ||
| ], | ||
| ) | ||
|
|
||
| # Don't depend on this directly; this is only used for the benchmark test | ||
| # generated by tf_library. | ||
| cc_library( | ||
| name = "tf_library_test_main", | ||
| testonly = 1, | ||
| visibility = ["//visibility:public"], | ||
| deps = ["//tensorflow/core:test_main"], | ||
| ) | ||
|
|
||
| xla_proto_library( | ||
| name = "tfcompile_proto", | ||
| srcs = ["tfcompile.proto"], | ||
| deps = [ | ||
| "//tensorflow/core:protos_all_cc", | ||
| ], | ||
| ) | ||
|
|
||
| cc_library( | ||
| name = "tfcompile_lib", | ||
| srcs = [ | ||
| "codegen.cc", | ||
| "compile.cc", | ||
| "flags.cc", | ||
| "tfcompile_util.cc", | ||
| ], | ||
| hdrs = [ | ||
| "codegen.h", | ||
| "compile.h", | ||
| "flags.h", | ||
| "tfcompile_util.h", | ||
| ], | ||
| deps = [ | ||
| ":runtime", # needed by codegen to print aligned_buffer_bytes | ||
| ":tfcompile_proto", | ||
| "//tensorflow/compiler/tf2xla:common", | ||
| "//tensorflow/compiler/tf2xla:xla_compiler", | ||
| "//tensorflow/compiler/tf2xla/kernels:xla_cpu_only_ops", | ||
| "//tensorflow/compiler/tf2xla/kernels:xla_ops", | ||
| "//tensorflow/compiler/xla:shape_util", | ||
| "//tensorflow/compiler/xla:statusor", | ||
| "//tensorflow/compiler/xla:util", | ||
| "//tensorflow/compiler/xla:xla_data_proto", | ||
| "//tensorflow/compiler/xla/client:client_library", | ||
| "//tensorflow/compiler/xla/client:local_client", | ||
| "//tensorflow/compiler/xla/service:compiler", | ||
| "//tensorflow/compiler/xla/service/cpu:cpu_compiler", | ||
| "//tensorflow/core:core_cpu", | ||
| "//tensorflow/core:core_cpu_internal", | ||
| "//tensorflow/core:framework", | ||
| "//tensorflow/core:framework_internal", | ||
| "//tensorflow/core:lib", | ||
| "//tensorflow/core:protos_all_cc", | ||
| "//tensorflow/core:stream_executor_no_cuda", | ||
| ], | ||
| ) | ||
|
|
||
| cc_test( | ||
| name = "codegen_test", | ||
| srcs = ["codegen_test.cc"], | ||
| data = ["codegen_test_h.golden"], | ||
| deps = [ | ||
| ":tfcompile_lib", | ||
| "//tensorflow/compiler/xla:shape_util", | ||
| "//tensorflow/core:lib", | ||
| "//tensorflow/core:test", | ||
| "//tensorflow/core:test_main", | ||
| ], | ||
| ) | ||
|
|
||
| cc_test( | ||
| name = "tfcompile_util_test", | ||
| srcs = ["tfcompile_util_test.cc"], | ||
| deps = [ | ||
| ":tfcompile_lib", | ||
| "//tensorflow/core:lib", | ||
| "//tensorflow/core:test", | ||
| "//tensorflow/core:test_main", | ||
| ], | ||
| ) | ||
|
|
||
| cc_binary( | ||
| name = "tfcompile", | ||
| visibility = ["//visibility:public"], | ||
| deps = [":tfcompile_main"], | ||
| ) | ||
|
|
||
| cc_library( | ||
| name = "tfcompile_main", | ||
| srcs = ["tfcompile_main.cc"], | ||
| visibility = ["//visibility:public"], | ||
| deps = [ | ||
| ":tfcompile_lib", | ||
| ":tfcompile_proto", | ||
| "//tensorflow/compiler/xla/legacy_flags:compiler_functor_flags", | ||
| "//tensorflow/compiler/xla/legacy_flags:cpu_compiler_flags", | ||
| "//tensorflow/compiler/xla/legacy_flags:cpu_runtime_flags", | ||
| "//tensorflow/compiler/xla/service:compiler", | ||
| "//tensorflow/core:core_cpu", | ||
| "//tensorflow/core:core_cpu_internal", | ||
| "//tensorflow/core:framework", | ||
| "//tensorflow/core:framework_internal", | ||
| "//tensorflow/core:lib", | ||
| "//tensorflow/core:protos_all_cc", | ||
| ], | ||
| ) | ||
|
|
||
| # NOTE: Most end-to-end tests are in the "tests" subdirectory, to ensure that | ||
| # tfcompile.bzl correctly handles usage from outside of the package that it is | ||
| # defined in. | ||
|
|
||
| # A simple test of tf_library from a text protobuf, mostly to enable the | ||
| # benchmark_test. | ||
| tf_library( | ||
| name = "test_graph_tfadd", | ||
| testonly = 1, | ||
| config = "test_graph_tfadd.config.pbtxt", | ||
| cpp_class = "AddComp", | ||
| graph = "test_graph_tfadd.pbtxt", | ||
| tags = ["manual"], | ||
| ) | ||
|
|
||
| # Utility library for benchmark binaries, used by the *_benchmark rules that are | ||
| # added by the tfcompile bazel macro. | ||
| cc_library( | ||
| name = "benchmark", | ||
| srcs = ["benchmark.cc"], | ||
| hdrs = ["benchmark.h"], | ||
| visibility = ["//visibility:public"], | ||
| deps = [ | ||
| # The purpose of the benchmark library is to support building an aot | ||
| # binary with minimal dependencies, to demonstrate small binary sizes. | ||
| # | ||
| # KEEP THE DEPENDENCIES MINIMAL. | ||
| "//tensorflow/core:framework_lite", | ||
| ], | ||
| ) | ||
|
|
||
| cc_library( | ||
| name = "benchmark_extra_android", | ||
| tags = [ | ||
| "manual", | ||
| "notap", | ||
| ], | ||
| visibility = ["//visibility:public"], | ||
| ) | ||
|
|
||
| cc_test( | ||
| name = "benchmark_test", | ||
| srcs = ["benchmark_test.cc"], | ||
| tags = ["manual"], | ||
| deps = [ | ||
| ":benchmark", | ||
| ":test_graph_tfadd", | ||
| "//tensorflow/core:test", | ||
| "//tensorflow/core:test_main", | ||
| ], | ||
| ) | ||
|
|
||
| test_suite( | ||
| name = "all_tests", | ||
| tags = ["manual"], | ||
| tests = [ | ||
| ":benchmark_test", | ||
| ":test_graph_tfadd_test", | ||
| "//tensorflow/compiler/aot/tests:all_tests", | ||
| ], | ||
| ) | ||
|
|
||
| exports_files([ | ||
| "benchmark_main.template", # used by tf_library(...,gen_benchmark=True) | ||
| "test.cc", # used by tf_library(...,gen_test=True) | ||
| ]) | ||
|
|
||
| # ----------------------------------------------------------------------------- | ||
|
|
||
| filegroup( | ||
| name = "all_files", | ||
| srcs = glob( | ||
| ["**/*"], | ||
| exclude = [ | ||
| "**/METADATA", | ||
| "**/OWNERS", | ||
| ], | ||
| ), | ||
| visibility = ["//tensorflow:__subpackages__"], | ||
| ) |
Oops, something went wrong.